Transformers 文档
GLM-4.1V
并获得增强的文档体验
开始使用
GLM-4.1V
以下示例演示了如何使用 Pipeline 或 AutoModel 类根据图像生成文本。
import torch
from transformers import pipeline
pipe = pipeline(
task="image-text-to-text",
model="THUDM/GLM-4.1V-9B-Thinking",
device=0,
torch_dtype=torch.bfloat16
)
messages = [
{
"role": "user",
"content": [
{
"type": "image",
"url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg",
},
{ "type": "text", "text": "Describe this image."},
]
}
]
pipe(text=messages,max_new_tokens=20, return_full_text=False)
将 GLM-4.1V 与视频输入一起使用与将其与图像输入一起使用类似。该模型可以处理视频数据并根据视频内容生成文本。
from transformers import AutoProcessor, Glm4vForConditionalGeneration
import torch
processor = AutoProcessor.from_pretrained("THUDM/GLM-4.1V-9B-Thinking")
model = Glm4vForConditionalGeneration.from_pretrained(
pretrained_model_name_or_path="THUDM/GLM-4.1V-9B-Thinking",
torch_dtype=torch.bfloat16,
device_map="cuda:0"
)
messages = [
{
"role": "user",
"content": [
{
"type": "video",
"url": "https://test-videos.co.uk/vids/bigbuckbunny/mp4/h264/720/Big_Buck_Bunny_720_10s_10MB.mp4",
},
{
"type": "text",
"text": "discribe this video",
},
],
}
]
inputs = processor.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_dict=True, return_tensors="pt", padding=True).to("cuda:0")
generated_ids = model.generate(**inputs, max_new_tokens=1024, do_sample=True, temperature=1.0)
output_text = processor.decode(generated_ids[0][inputs["input_ids"].shape[1] :], skip_special_tokens=True)
print(output_text)
Glm4vConfig
class transformers.Glm4vConfig
< 来源 >( text_config = None vision_config = None image_token_id = 151343 video_token_id = 151344 image_start_token_id = 151339 image_end_token_id = 151340 video_start_token_id = 151341 video_end_token_id = 151342 **kwargs )
参数
- text_config (
Union[PreTrainedConfig, dict]
, 可选, 默认为Glm4vTextConfig
) — 文本主干的配置对象或字典。 - vision_config (
Union[PreTrainedConfig, dict]
, 可选, 默认为Glm4vVisionConfig
) — 视觉主干的配置对象或字典。 - image_token_id (
int
, 可选, 默认为 151343) — 用于编码图像提示的图像令牌索引。 - video_token_id (
int
, 可选, 默认为 151344) — 用于编码图像提示的视频令牌索引。 - image_start_token_id (
int
, 可选, 默认为 151339) — 用于编码图像开头的图像开始令牌索引。 - image_end_token_id (
int
, 可选, 默认为 151340) — 用于编码图像末尾的图像结束令牌索引。 - video_start_token_id (
int
, 可选, 默认为 151341) — 用于编码视频开头的视频开始令牌索引。 - video_end_token_id (
int
, 可选, 默认为 151342) — 用于编码视频末尾的视频结束令牌索引。
这是用于存储 Glm4vModel 配置的配置类。它用于根据指定的参数实例化 GLM-4.1V 模型,定义模型架构。使用默认值实例化配置将生成与 GLM-4.1V-9B-Thinking THUDM/GLM-4.1V-9B-Thinking 类似的配置。
配置对象继承自 PretrainedConfig,可用于控制模型输出。有关更多信息,请参阅 PretrainedConfig 的文档。
>>> from transformers import Glm4vForConditionalGeneration, Glm4vConfig
>>> # Initializing a GLM-4.1V style configuration
>>> configuration = Glm4vConfig()
>>> # Initializing a model from the GLM-4.1V style configuration
>>> model = Glm4vForConditionalGeneration(configuration)
>>> # Accessing the model configuration
>>> configuration = model.config
Glm4vTextConfig
class transformers.Glm4vTextConfig
< 来源 >( vocab_size = 151552 hidden_size = 4096 intermediate_size = 13696 num_hidden_layers = 40 num_attention_heads = 32 num_key_value_heads = 2 hidden_act = 'silu' max_position_embeddings = 32768 initializer_range = 0.02 rms_norm_eps = 1e-05 use_cache = True tie_word_embeddings = False rope_theta = 10000.0 attention_dropout = 0.0 rope_scaling = None image_token_id = None video_token_id = None **kwargs )
参数
- vocab_size (
int
, 可选, 默认为 151552) — Glm4v 模型的词汇量。定义了调用 Glm4vModel 时传递的inputs_ids
可以表示的不同令牌的数量。 - hidden_size (
int
, 可选, 默认为 4096) — 隐藏表示的维度。 - intermediate_size (
int
, 可选, 默认为 13696) — MLP 表示的维度。 - num_hidden_layers (
int
, 可选, 默认为 40) — Transformer 编码器中的隐藏层数量。 - num_attention_heads (
int
, 可选, 默认为 32) — Transformer 编码器中每个注意力层的注意力头数量。 - num_key_value_heads (
int
, 可选, 默认为 2) — 用于实现分组查询注意力(Grouped Query Attention)的键值头数量。如果num_key_value_heads=num_attention_heads
,模型将使用多头注意力(MHA);如果num_key_value_heads=1
,模型将使用多查询注意力(MQA),否则使用 GQA。将多头检查点转换为 GQA 检查点时,每个分组键和值头应通过对该组内的所有原始头进行均值池化来构建。有关更多详细信息,请查看 此论文。如果未指定,将默认为32
。 - hidden_act (
str
或function
, 可选, 默认为"silu"
) — 解码器中的非线性激活函数(函数或字符串)。 - max_position_embeddings (
int
, 可选, 默认为 32768) — 此模型可能使用的最大序列长度。 - initializer_range (
float
, 可选, 默认为 0.02) — 用于初始化所有权重矩阵的 truncated_normal_initializer 的标准差。 - rms_norm_eps (
float
, 可选, 默认为 1e-05) — RMS 归一化层使用的 epsilon。 - use_cache (
bool
, 可选, 默认为True
) — 模型是否应返回最后一个键/值注意力(并非所有模型都使用)。仅在config.is_decoder=True
时相关。 - tie_word_embeddings (
bool
, 可选, 默认为False
) — 模型的输入和输出词嵌入是否应绑定。 - rope_theta (
float
, 可选, 默认为 10000.0) — RoPE 嵌入的基周期。 - attention_dropout (
float
, 可选, 默认为 0.0) — 注意力概率的 dropout 比率。 - rope_scaling (
Dict
, 可选) — 包含 RoPE 嵌入缩放配置的字典。注意:如果您应用新的 RoPE 类型并期望模型能在更长的max_position_embeddings
上工作,我们建议您相应地更新此值。预期内容:rope_type
(str
): 要使用的 RoPE 子变体。可以是 ['default', 'linear', 'dynamic', 'yarn', 'longrope', 'llama3'] 之一,其中 'default' 是原始 RoPE 实现。factor
(float
, 可选): 除 'default' 外的所有 RoPE 类型均使用。应用于 RoPE 嵌入的缩放因子。在大多数缩放类型中,x 的factor
将使模型能够处理长度为 x * 原始最大预训练长度的序列。original_max_position_embeddings
(int
, 可选): 与 'dynamic', 'longrope' 和 'llama3' 一起使用。预训练期间使用的原始最大位置嵌入。attention_factor
(float
, 可选): 与 'yarn' 和 'longrope' 一起使用。应用于注意力计算的缩放因子。如果未指定,则默认为实现建议的值,使用factor
字段推断建议值。 - image_token_id (
int
, 可选) — 用作图像嵌入占位符的令牌索引。 - video_token_id (
int
, 可选) — 用作视频嵌入占位符的令牌索引。
这是用于存储 Glm4vModel 配置的配置类。它用于根据指定的参数实例化 GLM-4.1V 模型,定义模型架构。使用默认值实例化配置将生成与 GLM-4.1V-9B-Thinking THUDM/GLM-4.1V-9B-Thinking 类似的配置。
配置对象继承自 PretrainedConfig,可用于控制模型输出。有关更多信息,请参阅 PretrainedConfig 的文档。
>>> from transformers import Glm4vTextModel, Glm4vConfig
>>> # Initializing a GLM-4.1V style configuration
>>> configuration = Glm4vConfig()
>>> # Initializing a model from the GLM-4.1V style configuration
>>> model = Glm4vTextModel(configuration)
>>> # Accessing the model configuration
>>> configuration = model.config
Glm4vImageProcessor
class transformers.Glm4vImageProcessor
< 来源 >( do_resize: bool = True size: typing.Optional[dict[str, int]] = None resample: Resampling = <Resampling.BICUBIC: 3> do_rescale: bool = True rescale_factor: typing.Union[int, float] = 0.00392156862745098 do_normalize: bool = True image_mean: typing.Union[float, list[float], NoneType] = None image_std: typing.Union[float, list[float], NoneType] = None do_convert_rgb: bool = True patch_size: int = 14 temporal_patch_size: int = 2 merge_size: int = 2 **kwargs )
参数
- do_resize (
bool
, 可选, 默认为True
) — 是否调整图像的(高度,宽度)尺寸。 - size (
Dict[str, int]
可选, 默认为{"shortest_edge" -- 112 * 112, "longest_edge": 28 * 28 * 15000}
): 调整大小后图像的(height, width)
尺寸。可通过preprocess
方法中的size
参数覆盖。可用选项包括:{"height": int, "width": int}
: 图像将调整为确切的大小(height, width)
。不保留纵横比。{"shortest_edge": int, "longest_edge": int}
: 图像将调整为最大尺寸,同时保持纵横比,并将最短边小于或等于shortest_edge
,最长边小于或等于longest_edge
。{"max_height": int, "max_width": int}
: 图像将调整为最大尺寸,同时保持纵横比,并将高度小于或等于max_height
,宽度小于或等于max_width
。
- resample (
PILImageResampling
, 可选, 默认为Resampling.BICUBIC
) — 调整图像大小时使用的重采样过滤器。 - do_rescale (
bool
, 可选, 默认为True
) — 是否按指定的比例rescale_factor
重新缩放图像。 - rescale_factor (
int
或float
, 可选, 默认为1/255
) — 如果重新缩放图像,则使用的缩放因子。 - do_normalize (
bool
, 可选, 默认为True
) — 是否对图像进行归一化。 - image_mean (
float
或List[float]
, 可选, 默认为[0.48145466, 0.4578275, 0.40821073]
) — 用于图像归一化的均值。这是一个浮点数或图像每个通道的浮点数列表。 - image_std (
float
或List[float]
, 可选, 默认为[0.26862954, 0.26130258, 0.27577711]
) — 用于图像归一化的标准差。这是一个浮点数或图像每个通道的浮点数列表。 - do_convert_rgb (
bool
, 可选, 默认为True
) — 是否将图像转换为 RGB 格式。 - patch_size (
int
, 可选, 默认为 14) — 视觉编码器的空间补丁大小。 - temporal_patch_size (
int
, 可选, 默认为 2) — 视觉编码器的时间补丁大小。 - merge_size (
int
, 可选, 默认为 2) — 视觉编码器到 LLM 编码器的合并大小。
构建一个 GLM-4V 图像处理器,它根据原始图像动态调整图像大小。
预处理
< 源 >( images: typing.Union[ForwardRef('PIL.Image.Image'), numpy.ndarray, ForwardRef('torch.Tensor'), list['PIL.Image.Image'], list[numpy.ndarray], list['torch.Tensor']] videos: typing.Union[list['PIL.Image.Image'], ForwardRef('np.ndarray'), ForwardRef('torch.Tensor'), list['np.ndarray'], list['torch.Tensor'], list[list['PIL.Image.Image']], list[list['np.ndarrray']], list[list['torch.Tensor']]] = None do_resize: typing.Optional[bool] = None size: typing.Optional[dict[str, int]] = None resample: Resampling = None do_rescale: typing.Optional[bool] = None rescale_factor: typing.Optional[float] = None do_normalize: typing.Optional[bool] = None image_mean: typing.Union[float, list[float], NoneType] = None image_std: typing.Union[float, list[float], NoneType] = None patch_size: typing.Optional[int] = None temporal_patch_size: typing.Optional[int] = None merge_size: typing.Optional[int] = None do_convert_rgb: typing.Optional[bool] = None return_tensors: typing.Union[str, transformers.utils.generic.TensorType, NoneType] = None data_format: typing.Optional[transformers.image_utils.ChannelDimension] = <ChannelDimension.FIRST: 'channels_first'> input_data_format: typing.Union[str, transformers.image_utils.ChannelDimension, NoneType] = None )
参数
- images (
ImageInput
) — 待预处理的图像。期望像素值范围为 0 到 255 的单张或批处理图像。如果传入像素值范围在 0 到 1 之间的图像,请设置do_rescale=False
。 - videos (
VideoInput
) — 待预处理的视频。期望像素值范围为 0 到 255 的单个或批处理视频。如果传入像素值范围在 0 到 1 之间的视频,请设置do_rescale=False
。 - do_resize (
bool
, 可选, 默认为self.do_resize
) — 是否调整图像大小。 - size (
Dict[str, int]
, 可选, 默认为self.size
) — 调整大小后的图像大小。图像的最短边将调整为 size["shortest_edge"],最长边将保持输入宽高比进行调整。 - resample (
int
, 可选, 默认为self.resample
) — 如果调整图像大小,要使用的重采样滤镜。可以是枚举PILImageResampling
之一。仅当do_resize
设置为True
时有效。 - do_rescale (
bool
, 可选, 默认为self.do_rescale
) — 是否重新缩放图像。 - rescale_factor (
float
, 可选, 默认为self.rescale_factor
) — 如果do_rescale
设置为True
,则用于重新缩放图像的缩放因子。 - do_normalize (
bool
, 可选, 默认为self.do_normalize
) — 是否归一化图像。 - image_mean (
float
或List[float]
, 可选, 默认为self.image_mean
) — 用于归一化的图像均值。仅当do_normalize
设置为True
时有效。 - image_std (
float
或List[float]
, 可选, 默认为self.image_std
) — 用于归一化的图像标准差。仅当do_normalize
设置为True
时有效。要调整图像大小的最大像素数。 - patch_size (
int
, 可选, 默认为self.patch_size
) — 视觉编码器的空间补丁大小。 - temporal_patch_size (
int
, 可选, 默认为self.temporal_patch_size
) — 视觉编码器的时间补丁大小。 - merge_size (
int
, 可选, 默认为self.merge_size
) — 视觉编码器到 LLM 编码器的合并大小。 - do_convert_rgb (
bool
, 可选, 默认为self.do_convert_rgb
) — 是否将图像转换为 RGB 格式。 - return_tensors (
str
或TensorType
, 可选) — 要返回的张量类型。可以是以下之一:- 未设置:返回
np.ndarray
列表。 TensorType.TENSORFLOW
或'tf'
:返回tf.Tensor
类型的批处理。TensorType.PYTORCH
或'pt'
:返回torch.Tensor
类型的批处理。TensorType.NUMPY
或'np'
:返回np.ndarray
类型的批处理。TensorType.JAX
或'jax'
:返回jax.numpy.ndarray
类型的批处理。
- 未设置:返回
- data_format (
ChannelDimension
或str
, 可选, 默认为ChannelDimension.FIRST
) — 输出图像的通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
:图像格式为 (num_channels, height, width)。"channels_last"
或ChannelDimension.LAST
:图像格式为 (height, width, num_channels)。- 未设置:使用输入图像的通道维度格式。
- input_data_format (
ChannelDimension
或str
, 可选) — 输入图像的通道维度格式。如果未设置,则从输入图像推断通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
:图像格式为 (num_channels, height, width)。"channels_last"
或ChannelDimension.LAST
:图像格式为 (height, width, num_channels)。"none"
或ChannelDimension.NONE
:图像格式为 (height, width)。
Glm4vVideoProcessor
class transformers.Glm4vVideoProcessor
< 源 >( **kwargs: typing_extensions.Unpack[transformers.models.glm4v.video_processing_glm4v.Glm4vVideoProcessorInitKwargs] )
参数
- do_resize (
bool
, 可选, 默认为self.do_resize
) — 是否将视频的 (height, width) 维度调整到指定的size
。可在preprocess
方法中的do_resize
参数覆盖。 - size (
dict
, 可选, 默认为self.size
) — 调整大小后的输出视频的大小。可在preprocess
方法中的size
参数覆盖。 - size_divisor (
int
, 可选, 默认为self.size_divisor
) — 确保高度和宽度都可以被其整除的大小。 - default_to_square (
bool
, 可选, 默认为self.default_to_square
) — 如果大小为整数,是否在调整视频大小时默认为方形视频。 - resample (
PILImageResampling
, 可选, 默认为self.resample
) — 如果调整视频大小,要使用的重采样滤镜。仅当do_resize
设置为True
时有效。可在preprocess
方法中的resample
参数覆盖。 - do_center_crop (
bool
, 可选, 默认为self.do_center_crop
) — 是否将视频中心裁剪到指定的crop_size
。可在preprocess
方法中的do_center_crop
覆盖。 - do_pad (
bool
, 可选) — 是否将视频填充到批处理中视频的(max_height, max_width)
。 - crop_size (
dict[str, int]
可选, 默认为self.crop_size
) — 应用center_crop
后的输出视频大小。可在preprocess
方法中的crop_size
覆盖。 - do_rescale (
bool
, 可选, 默认为self.do_rescale
) — 是否按指定的比例rescale_factor
重新缩放视频。可在preprocess
方法中的do_rescale
参数覆盖。 - rescale_factor (
int
或float
, 可选, 默认为self.rescale_factor
) — 如果do_rescale
设置为True
,则用于重新缩放视频的缩放因子。可在preprocess
方法中的rescale_factor
参数覆盖。 - do_normalize (
bool
, 可选, 默认为self.do_normalize
) — 是否归一化视频。可在preprocess
方法中的do_normalize
参数覆盖。可在preprocess
方法中的do_normalize
参数覆盖。 - image_mean (
float
或list[float]
, 可选, 默认为self.image_mean
) — 用于视频归一化的均值。这是一个浮点数或视频中通道数长度的浮点数列表。可在preprocess
方法中的image_mean
参数覆盖。可在preprocess
方法中的image_mean
参数覆盖。 - image_std (
float
或list[float]
, 可选, 默认为self.image_std
) — 用于视频归一化的标准差。这是一个浮点数或视频中通道数长度的浮点数列表。可在preprocess
方法中的image_std
参数覆盖。可在preprocess
方法中的image_std
参数覆盖。 - do_convert_rgb (
bool
, 可选, 默认为self.image_std
) — 是否将视频转换为 RGB 格式。 - video_metadata (
VideoMetadata
, 可选) — 视频的元数据,包含总时长、帧率和总帧数信息。 - do_sample_frames (
int
, 可选, 默认为self.do_sample_frames
) — 是否在处理前从视频中采样帧或处理整个视频。 - num_frames (
int
, 可选, 默认为self.num_frames
) — 当do_sample_frames=True
时要采样的最大帧数。 - fps (
int
, 可选, 默认为self.fps
) — 当do_sample_frames=True
时,每秒采样的目标帧数。 - return_tensors (
str
或TensorType
, 可选) — 如果设置为 `pt` 则返回堆叠的张量,否则返回张量列表。 - data_format (
ChannelDimension
或str
, 可选, 默认为ChannelDimension.FIRST
) — 输出视频的通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
:视频格式为 (num_channels, height, width)。"channels_last"
或ChannelDimension.LAST
:视频格式为 (height, width, num_channels)。- 未设置:使用输入视频的通道维度格式。
- input_data_format (
ChannelDimension
或str
, 可选) — 输入视频的通道维度格式。如果未设置,则从输入视频推断通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
:视频格式为 (num_channels, height, width)。"channels_last"
或ChannelDimension.LAST
:视频格式为 (height, width, num_channels)。"none"
或ChannelDimension.NONE
:视频格式为 (height, width)。
- device (
torch.device
, 可选) — 处理视频的设备。如果未设置,则从输入视频推断设备。 - patch_size (
int
, 可选, 默认为 14) — 视觉编码器的空间补丁大小。 - temporal_patch_size (
int
, 可选, 默认为 2) — 视觉编码器的时间补丁大小。 - merge_size (
int
, 可选, 默认为 2) — 视觉编码器到 LLM 编码器的合并大小。
构建一个快速 GLM-4V 图像处理器,它根据原始视频动态调整视频大小。
预处理
< 源 >( videos: typing.Union[list['PIL.Image.Image'], ForwardRef('np.ndarray'), ForwardRef('torch.Tensor'), list['np.ndarray'], list['torch.Tensor'], list[list['PIL.Image.Image']], list[list['np.ndarrray']], list[list['torch.Tensor']]] **kwargs: typing_extensions.Unpack[transformers.processing_utils.VideosKwargs] )
参数
- do_resize (
bool
, 可选, 默认为self.do_resize
) — 是否将视频的 (height, width) 维度调整到指定的size
。可在preprocess
方法中的do_resize
参数覆盖。 - size (
dict
, 可选, 默认为self.size
) — 调整大小后的输出视频的大小。可在preprocess
方法中的size
参数覆盖。 - size_divisor (
int
, 可选, 默认为self.size_divisor
) — 确保高度和宽度都可以被其整除的大小。 - default_to_square (
bool
, 可选, 默认为self.default_to_square
) — 如果大小为整数,是否在调整视频大小时默认为方形视频。 - resample (
PILImageResampling
, 可选, 默认为self.resample
) — 如果调整视频大小,要使用的重采样滤镜。仅当do_resize
设置为True
时有效。可在preprocess
方法中的resample
参数覆盖。 - do_center_crop (
bool
, 可选, 默认为self.do_center_crop
) — 是否将视频中心裁剪到指定的crop_size
。可在preprocess
方法中的do_center_crop
覆盖。 - do_pad (
bool
, 可选) — 是否将视频填充到批处理中视频的(max_height, max_width)
。 - crop_size (
dict[str, int]
可选, 默认为self.crop_size
) — 应用center_crop
后的输出视频大小。可在preprocess
方法中的crop_size
覆盖。 - do_rescale (
bool
, 可选, 默认为self.do_rescale
) — 是否按指定的比例rescale_factor
重新缩放视频。可在preprocess
方法中的do_rescale
参数覆盖。 - rescale_factor (
int
或float
, 可选, 默认为self.rescale_factor
) — 如果对视频进行重新缩放,则使用的缩放因子。仅当do_rescale
设置为True
时才有效。可以通过preprocess
方法中的rescale_factor
参数覆盖。 - do_normalize (
bool
, 可选, 默认为self.do_normalize
) — 是否对视频进行归一化。可以通过preprocess
方法中的do_normalize
参数覆盖。可以通过preprocess
方法中的do_normalize
参数覆盖。 - image_mean (
float
或list[float]
, 可选, 默认为self.image_mean
) — 如果对视频进行归一化,则使用的均值。这是一个浮点数或浮点数列表,其长度与视频中的通道数相同。可以通过preprocess
方法中的image_mean
参数覆盖。可以通过preprocess
方法中的image_mean
参数覆盖。 - image_std (
float
或list[float]
, 可选, 默认为self.image_std
) — 如果对视频进行归一化,则使用的标准差。这是一个浮点数或浮点数列表,其长度与视频中的通道数相同。可以通过preprocess
方法中的image_std
参数覆盖。可以通过preprocess
方法中的image_std
参数覆盖。 - do_convert_rgb (
bool
, 可选, 默认为self.image_std
) — 是否将视频转换为 RGB。 - video_metadata (
VideoMetadata
, 可选) — 视频的元数据,包含总时长、每秒帧数和总帧数信息。 - do_sample_frames (
int
, 可选, 默认为self.do_sample_frames
) — 是在处理前从视频中采样帧,还是处理整个视频。 - num_frames (
int
, 可选, 默认为self.num_frames
) — 当do_sample_frames=True
时,要采样的最大帧数。 - fps (
int
, 可选, 默认为self.fps
) — 当do_sample_frames=True
时,每秒采样的目标帧数。 - return_tensors (
str
或TensorType
, 可选) — 如果设置为 `pt`,则返回堆叠的张量,否则返回张量列表。 - data_format (
ChannelDimension
或str
, 可选, 默认为ChannelDimension.FIRST
) — 输出视频的通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
: 视频格式为 (通道数, 高度, 宽度)。"channels_last"
或ChannelDimension.LAST
: 视频格式为 (高度, 宽度, 通道数)。- 未设置: 使用输入视频的通道维度格式。
- input_data_format (
ChannelDimension
或str
, 可选) — 输入视频的通道维度格式。如果未设置,则从输入视频推断通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
: 视频格式为 (通道数, 高度, 宽度)。"channels_last"
或ChannelDimension.LAST
: 视频格式为 (高度, 宽度, 通道数)。"none"
或ChannelDimension.NONE
: 视频格式为 (高度, 宽度)。
- device (
torch.device
, 可选) — 处理视频的设备。如果未设置,则从输入视频推断设备。
Glm4vImageProcessorFast
class transformers.Glm4vImageProcessorFast
< 来源 >( **kwargs: typing_extensions.Unpack[transformers.models.glm4v.image_processing_glm4v_fast.Glm4vFastImageProcessorKwargs] )
构造一个快速 Glm4V 图像处理器。
预处理
< 来源 >( images: typing.Union[ForwardRef('PIL.Image.Image'), numpy.ndarray, ForwardRef('torch.Tensor'), list['PIL.Image.Image'], list[numpy.ndarray], list['torch.Tensor']] videos: typing.Union[list['PIL.Image.Image'], ForwardRef('np.ndarray'), ForwardRef('torch.Tensor'), list['np.ndarray'], list['torch.Tensor'], list[list['PIL.Image.Image']], list[list['np.ndarrray']], list[list['torch.Tensor']]] = None do_resize: typing.Optional[bool] = None size: typing.Optional[dict[str, int]] = None resample: typing.Union[ForwardRef('PILImageResampling'), ForwardRef('F.InterpolationMode'), NoneType] = None do_rescale: typing.Optional[bool] = None rescale_factor: typing.Optional[float] = None do_normalize: typing.Optional[bool] = None image_mean: typing.Union[float, list[float], NoneType] = None image_std: typing.Union[float, list[float], NoneType] = None patch_size: typing.Optional[int] = None temporal_patch_size: typing.Optional[int] = None merge_size: typing.Optional[int] = None do_convert_rgb: typing.Optional[bool] = None return_tensors: typing.Union[str, transformers.utils.generic.TensorType, NoneType] = None data_format: typing.Optional[transformers.image_utils.ChannelDimension] = <ChannelDimension.FIRST: 'channels_first'> input_data_format: typing.Union[str, transformers.image_utils.ChannelDimension, NoneType] = None device: typing.Optional[ForwardRef('torch.device')] = None disable_grouping: typing.Optional[bool] = False **kwargs )
参数
- images (
Union[PIL.Image.Image, numpy.ndarray, torch.Tensor, list['PIL.Image.Image'], list[numpy.ndarray], list['torch.Tensor']]
) — 要预处理的图像。期望单个或批次图像,像素值范围为 0 到 255。如果传入的图像像素值在 0 到 1 之间,请将do_rescale=False
。 - videos (
Union[list['PIL.Image.Image'], np.ndarray, torch.Tensor, list['np.ndarray'], list['torch.Tensor'], list[list['PIL.Image.Image']], list[list['np.ndarrray']], list[list['torch.Tensor']]]
) — 要预处理的视频。期望单个或批次视频,像素值范围为 0 到 255。如果传入的视频像素值在 0 到 1 之间,请将do_rescale=False
。 - do_resize (
bool
, 可选) — 是否调整图像大小。 - size (
dict[str, int]
, 可选) — 描述模型最大输入尺寸。 - resample (
Union[PILImageResampling, F.InterpolationMode, NoneType]
) — 如果调整图像大小,则使用的重采样滤波器。这可以是枚举PILImageResampling
之一。仅当do_resize
设置为True
时才有效。 - do_rescale (
bool
, 可选) — 是否重新缩放图像。 - rescale_factor (
float
, 可选) — 如果do_rescale
设置为True
,则按此缩放因子对图像进行重新缩放。 - do_normalize (
bool
, 可选) — 是否对图像进行归一化。 - image_mean (
Union[float, list[float], NoneType]
) — 用于归一化的图像均值。仅当do_normalize
设置为True
时才有效。 - image_std (
Union[float, list[float], NoneType]
) — 用于归一化的图像标准差。仅当do_normalize
设置为True
时才有效。 - patch_size (
int
, 可选, 默认为 14) — 视觉编码器的空间补丁大小。 - temporal_patch_size (
int
, 可选, 默认为 2) — 视觉编码器的时间补丁大小。 - merge_size (
int
, 可选, 默认为 2) — 视觉编码器到 LLM 编码器的合并大小。 - do_convert_rgb (
bool
, 可选) — 是否将图像转换为 RGB。 - return_tensors (
Union[str, ~utils.generic.TensorType, NoneType]
) — 如果设置为 `pt`,则返回堆叠的张量,否则返回张量列表。 - data_format (
~image_utils.ChannelDimension
, 可选, 默认为ChannelDimension.FIRST
) — 仅支持ChannelDimension.FIRST
。为兼容慢速处理器而添加。 - input_data_format (
Union[str, ~image_utils.ChannelDimension, NoneType]
) — 输入图像的通道维度格式。如果未设置,则从输入图像推断通道维度格式。可以是以下之一:"channels_first"
或ChannelDimension.FIRST
: 图像格式为 (通道数, 高度, 宽度)。"channels_last"
或ChannelDimension.LAST
: 图像格式为 (高度, 宽度, 通道数)。"none"
或ChannelDimension.NONE
: 图像格式为 (高度, 宽度)。
- device (
torch.device
, 可选) — 处理图像的设备。如果未设置,则从输入图像推断设备。 - disable_grouping (
bool
, 可选, 默认为False
) — 是否禁用按大小对图像进行分组,以单独处理而非批量处理。如果为 None,则如果图像在 CPU 上,则设置为 True,否则设置为 False。此选择基于经验观察,详情请参阅:https://github.com/huggingface/transformers/pull/38157
Glm4vProcessor
class transformers.Glm4vProcessor
< 来源 >( image_processor = None tokenizer = None video_processor = None chat_template = None **kwargs )
参数
- image_processor (Glm4vProcessor, 可选) — 图像处理器是必需输入。
- tokenizer (PreTrainedTokenizerFast, 可选) — 分词器是必需输入。
- video_processor (Glm4vVideoProcessor, 可选) — 视频处理器是必需输入。
- chat_template (
str
, 可选) — 一个 Jinja 模板,用于将聊天中的消息列表转换为可分词字符串。
构造一个 GLM-4V 处理器,它将 GLM-4V 图像处理器和 GLM-4 分词器封装到一个单独的处理器中。有关更多信息,请参阅 __call__()
和 decode()。
此方法将其所有参数转发到 Qwen2TokenizerFast 的 batch_decode()。有关更多信息,请参阅此方法的文档字符串。
post_process_image_text_to_text
< 来源 >( generated_outputs skip_special_tokens = True clean_up_tokenization_spaces = False **kwargs ) → list[str]
参数
- generated_outputs (
torch.Tensor
或np.ndarray
) — 模型generate
函数的输出。输出应为形状为(batch_size, sequence_length)
或(sequence_length,)
的张量。 - skip_special_tokens (
bool
, 可选, 默认为True
) — 是否在输出中移除特殊 token。参数传递给分词器的batch_decode
方法。 - clean_up_tokenization_spaces (
bool
, 可选, 默认为False
) — 是否清理分词空间。参数传递给分词器的batch_decode
方法。 - **kwargs — 要传递给分词器
batch_decode
方法的其他参数。
返回
list[str]
解码后的文本。
后处理模型输出以解码文本。
Glm4vTextModel
class transformers.Glm4vTextModel
< 来源 >( config: Glm4vTextConfig )
参数
- config (Glm4vTextConfig) — 模型配置类,包含模型的所有参数。使用配置文件初始化不加载与模型关联的权重,仅加载配置。请查看 from_pretrained() 方法加载模型权重。
裸 Glm4V 文本模型,输出原始隐藏状态,不带任何特定头部。
此模型继承自 PreTrainedModel。请查看超类文档,了解库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、修剪头部等)。
此模型也是 PyTorch torch.nn.Module 子类。将其作为常规 PyTorch 模块使用,并参考 PyTorch 文档了解所有与一般使用和行为相关的事项。
forward
< 来源 >( input_ids: typing.Optional[torch.LongTensor] = None attention_mask: typing.Optional[torch.Tensor] = None position_ids: typing.Optional[torch.LongTensor] = None past_key_values: typing.Optional[list[torch.FloatTensor]] = None inputs_embeds: typing.Optional[torch.FloatTensor] = None use_cache: typing.Optional[bool] = None output_attentions: typing.Optional[bool] = None output_hidden_states: typing.Optional[bool] = None cache_position: typing.Optional[torch.LongTensor] = None **kwargs: typing_extensions.Unpack[transformers.modeling_flash_attention_utils.FlashAttentionKwargs] ) → transformers.modeling_outputs.BaseModelOutputWithPast 或 tuple(torch.FloatTensor)
参数
- input_ids (形状为
(batch_size, sequence_length)
的torch.LongTensor
, 可选) — 词汇表中输入序列 token 的索引。填充默认将被忽略。索引可以使用 AutoTokenizer 获取。有关详细信息,请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。
- attention_mask (形状为
(batch_size, sequence_length)
的torch.Tensor
, 可选) — 掩码,用于避免在填充 token 索引上执行注意力。掩码值选择在[0, 1]
之间:- 1 表示**未被掩码**的 token,
- 0 表示**被掩码**的 token。
- position_ids (形状为
(batch_size, sequence_length)
的torch.LongTensor
, 可选) — 位置嵌入中每个输入序列 token 位置的索引。在[0, config.n_positions - 1]
范围内选择。 - past_key_values (
list[torch.FloatTensor]
, 可选) — 预计算的隐藏状态(自注意力块和交叉注意力块中的键和值),可用于加速顺序解码。这通常包括模型在解码前期返回的past_key_values
,当use_cache=True
或config.use_cache=True
时。允许两种格式:
- 一个 Cache 实例,详见我们的 kv cache 指南;
- 一个长度为
config.n_layers
的tuple(torch.FloatTensor)
元组,每个元组包含两个形状为(batch_size, num_heads, sequence_length, embed_size_per_head)
的张量)。这也称为传统缓存格式。
模型将输出与输入相同的缓存格式。如果未传入
past_key_values
,则将返回传统缓存格式。如果使用
past_key_values
,用户可以选择只输入形状为(batch_size, 1)
的最后一个input_ids
(那些没有将过去键值状态提供给此模型的),而不是形状为(batch_size, sequence_length)
的所有input_ids
。 - inputs_embeds (形状为
(batch_size, sequence_length, hidden_size)
的torch.FloatTensor
,可选) — 可选地,您可以选择直接传递嵌入表示,而不是传递input_ids
。如果您想对如何将input_ids
索引转换为关联向量进行比模型内部嵌入查找矩阵更精细的控制,这会很有用。 - use_cache (
bool
, 可选) — 如果设置为True
,则返回past_key_values
键值状态,可用于加速解码(参见past_key_values
)。 - output_attentions (
bool
, 可选) — 是否返回所有注意力层的注意力张量。更多详情请参阅返回张量中的attentions
。 - output_hidden_states (
bool
, 可选) — 是否返回所有层的隐藏状态。更多详情请参阅返回张量中的hidden_states
。 - cache_position (形状为
(sequence_length)
的torch.LongTensor
,可选) — 描述输入序列令牌在序列中位置的索引。与position_ids
不同,此张量不受填充影响。它用于在正确位置更新缓存并推断完整的序列长度。
返回
transformers.modeling_outputs.BaseModelOutputWithPast 或 tuple(torch.FloatTensor)
一个 transformers.modeling_outputs.BaseModelOutputWithPast 或一个 torch.FloatTensor
的元组(如果传入 return_dict=False
或当 config.return_dict=False
时),包含根据配置(None
)和输入的不同元素。
-
last_hidden_state (
torch.FloatTensor
, 形状为(batch_size, sequence_length, hidden_size)
) — 模型最后一层输出的隐藏状态序列。如果使用了
past_key_values
,则只输出形状为(batch_size, 1, hidden_size)
的序列的最后一个隐藏状态。 -
past_key_values (
Cache
, 可选, 当传入use_cache=True
或当config.use_cache=True
时返回) — 这是一个 Cache 实例。更多详情,请参阅我们的 kv cache 指南。包含预计算的隐藏状态(自注意力块中的键和值,如果
config.is_encoder_decoder=True
则可选地包含交叉注意力块中的键和值),可用于(参见past_key_values
输入)加速顺序解码。 -
hidden_states (
tuple(torch.FloatTensor)
, 可选, 当传入output_hidden_states=True
或当config.output_hidden_states=True
时返回) —torch.FloatTensor
的元组(如果模型有嵌入层,则一个用于嵌入层的输出,加上每个层的一个输出),形状为(batch_size, sequence_length, hidden_size)
。模型在每个层输出的隐藏状态以及可选的初始嵌入输出。
-
attentions (
tuple(torch.FloatTensor)
, 可选, 当传入output_attentions=True
或当config.output_attentions=True
时返回) —torch.FloatTensor
的元组(每个层一个),形状为(batch_size, num_heads, sequence_length, sequence_length)
。注意力 softmax 后的注意力权重,用于计算自注意力头中的加权平均值。
Glm4vTextModel 的前向方法,覆盖了 __call__
特殊方法。
尽管前向传播的配方需要在该函数中定义,但之后应该调用 Module
实例而不是此函数,因为前者负责运行预处理和后处理步骤,而后者则默默地忽略它们。
Glm4vModel
class transformers.Glm4vModel
< source >( config )
参数
- config (Glm4vModel) — 包含模型所有参数的模型配置类。使用配置文件初始化不加载与模型关联的权重,只加载配置。请查阅 from_pretrained() 方法以加载模型权重。
裸 Glm4V 模型,输出原始隐藏状态,顶部没有任何特定头部。
此模型继承自 PreTrainedModel。请查看超类文档,了解库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、修剪头部等)。
此模型也是 PyTorch torch.nn.Module 子类。将其作为常规 PyTorch 模块使用,并参考 PyTorch 文档了解所有与一般使用和行为相关的事项。
forward
< source >( input_ids: LongTensor = None attention_mask: typing.Optional[torch.Tensor] = None position_ids: typing.Optional[torch.LongTensor] = None past_key_values: typing.Optional[list[torch.FloatTensor]] = None inputs_embeds: typing.Optional[torch.FloatTensor] = None use_cache: typing.Optional[bool] = None output_attentions: typing.Optional[bool] = None output_hidden_states: typing.Optional[bool] = None pixel_values: typing.Optional[torch.Tensor] = None pixel_values_videos: typing.Optional[torch.FloatTensor] = None image_grid_thw: typing.Optional[torch.LongTensor] = None video_grid_thw: typing.Optional[torch.LongTensor] = None rope_deltas: typing.Optional[torch.LongTensor] = None cache_position: typing.Optional[torch.LongTensor] = None **kwargs: typing_extensions.Unpack[transformers.models.glm4v.modeling_glm4v.KwargsForCausalLM] ) → transformers.models.glm4v.modeling_glm4v.Glm4vModelOutputWithPast
或 tuple(torch.FloatTensor)
参数
- input_ids (形状为
(batch_size, sequence_length)
的torch.LongTensor
) — 词汇表中输入序列令牌的索引。默认情况下将忽略填充。索引可以使用 AutoTokenizer 获取。有关详情,请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。
- attention_mask (形状为
(batch_size, sequence_length)
的torch.Tensor
,可选) — 避免对填充令牌索引执行注意力的掩码。掩码值选择在[0, 1]
之间:- 1 表示未被掩盖的令牌,
- 0 表示被掩盖的令牌。
- position_ids (形状为
(batch_size, sequence_length)
的torch.LongTensor
,可选) — 每个输入序列令牌在位置嵌入中的位置索引。选择范围为[0, config.n_positions - 1]
。 - past_key_values (
list[torch.FloatTensor]
, 可选) — 预计算的隐藏状态(自注意力块和交叉注意力块中的键和值),可用于加速顺序解码。这通常包括模型在解码前期返回的past_key_values
,当use_cache=True
或config.use_cache=True
时。允许两种格式:
- 一个 Cache 实例,详见我们的 kv cache 指南;
- 一个长度为
config.n_layers
的tuple(torch.FloatTensor)
元组,每个元组包含两个形状为(batch_size, num_heads, sequence_length, embed_size_per_head)
的张量)。这也称为传统缓存格式。
模型将输出与输入相同的缓存格式。如果未传入
past_key_values
,则将返回传统缓存格式。如果使用
past_key_values
,用户可以选择只输入形状为(batch_size, 1)
的最后一个input_ids
(那些没有将过去键值状态提供给此模型的),而不是形状为(batch_size, sequence_length)
的所有input_ids
。 - inputs_embeds (形状为
(batch_size, sequence_length, hidden_size)
的torch.FloatTensor
,可选) — 可选地,您可以选择直接传递嵌入表示,而不是传递input_ids
。如果您想对如何将input_ids
索引转换为关联向量进行比模型内部嵌入查找矩阵更精细的控制,这会很有用。 - use_cache (
bool
, 可选) — 如果设置为True
,则返回past_key_values
键值状态,可用于加速解码(参见past_key_values
)。 - output_attentions (
bool
, 可选) — 是否返回所有注意力层的注意力张量。更多详情请参阅返回张量中的attentions
。 - output_hidden_states (
bool
, 可选) — 是否返回所有层的隐藏状态。更多详情请参阅返回张量中的hidden_states
。 - pixel_values (形状为
(batch_size, num_channels, image_size, image_size)
的torch.Tensor
,可选) — 对应输入图像的张量。像素值可以使用{image_processor_class}
获取。有关详情,请参阅{image_processor_class}.__call__
({processor_class}
使用{image_processor_class}
处理图像)。 - pixel_values_videos (形状为 `(seq_length, num_channels * temporal_size * image_size * image_size)` 的
torch.FloatTensor
,可选) — 对应输入视频的张量。像素值可以使用 AutoImageProcessor 获取。有关详情,请参阅 Glm4vImageProcessor.call()。Glm4vProcessor 使用 Glm4vImageProcessor 处理视频。 - image_grid_thw (形状为
(num_images, 3)
的torch.LongTensor
,可选) — LLM 中每张图像的特征形状的时间、高度和宽度。 - video_grid_thw (形状为
(num_videos, 3)
的torch.LongTensor
,可选) — LLM 中每个视频的特征形状的时间、高度和宽度。 - rope_deltas (形状为
(batch_size, )
的torch.LongTensor
,可选) — 序列长度与多模态旋转位置编码之间的旋转位置编码索引差。 - cache_position (形状为
(sequence_length)
的torch.LongTensor
,可选) — 描述输入序列令牌在序列中位置的索引。与position_ids
不同,此张量不受填充影响。它用于在正确位置更新缓存并推断完整的序列长度。
返回
transformers.models.glm4v.modeling_glm4v.Glm4vModelOutputWithPast
或 tuple(torch.FloatTensor)
一个 transformers.models.glm4v.modeling_glm4v.Glm4vModelOutputWithPast
或一个 torch.FloatTensor
的元组(如果传入 return_dict=False
或当 config.return_dict=False
时),包含根据配置(None
)和输入的不同元素。
-
last_hidden_state (形状为
(batch_size, sequence_length, hidden_size)
的<class 'torch.FloatTensor'>.last_hidden_state
,默认为None
) — 模型最后一层的隐藏状态序列。 -
past_key_values (
tuple(tuple(torch.FloatTensor))
, 可选, 当传入use_cache=True
或当config.use_cache=True
时返回) — 长度为config.n_layers
的tuple(torch.FloatTensor)
元组,每个元组包含 2 个形状为(batch_size, num_heads, sequence_length, embed_size_per_head)
的张量。包含预计算的隐藏状态(自注意力块中的键和值),可用于(参见
past_key_values
输入)加速顺序解码。 -
hidden_states (
tuple[torch.FloatTensor]
, 可选, 当传入output_hidden_states=True
或当config.output_hidden_states=True
时返回) —torch.FloatTensor
的元组(如果模型有嵌入层,则一个用于嵌入层的输出,加上每个层的一个输出),形状为(batch_size, sequence_length, hidden_size)
。模型在每个层输出的隐藏状态以及可选的初始嵌入输出。
-
attentions (
tuple[torch.FloatTensor]
, 可选, 当传入output_attentions=True
或当config.output_attentions=True
时返回) —torch.FloatTensor
的元组(每个层一个),形状为(batch_size, num_heads, sequence_length, sequence_length)
。注意力 softmax 后的注意力权重,用于计算自注意力头中的加权平均值。
-
rope_deltas (形状为
(batch_size, )
的torch.LongTensor
,可选) — 序列长度与多模态旋转位置编码之间的旋转位置编码索引差。
Glm4vModel 的前向方法,覆盖了 __call__
特殊方法。
尽管前向传播的配方需要在该函数中定义,但之后应该调用 Module
实例而不是此函数,因为前者负责运行预处理和后处理步骤,而后者则默默地忽略它们。
Glm4vForConditionalGeneration
forward
< source >( input_ids: LongTensor = None attention_mask: typing.Optional[torch.Tensor] = None position_ids: typing.Optional[torch.LongTensor] = None past_key_values: typing.Optional[list[torch.FloatTensor]] = None inputs_embeds: typing.Optional[torch.FloatTensor] = None labels: typing.Optional[torch.LongTensor] = None use_cache: typing.Optional[bool] = None output_attentions: typing.Optional[bool] = None output_hidden_states: typing.Optional[bool] = None return_dict: typing.Optional[bool] = None pixel_values: typing.Optional[torch.Tensor] = None pixel_values_videos: typing.Optional[torch.FloatTensor] = None image_grid_thw: typing.Optional[torch.LongTensor] = None video_grid_thw: typing.Optional[torch.LongTensor] = None rope_deltas: typing.Optional[torch.LongTensor] = None cache_position: typing.Optional[torch.LongTensor] = None **kwargs: typing_extensions.Unpack[transformers.models.glm4v.modeling_glm4v.KwargsForCausalLM] ) → transformers.models.glm4v.modeling_glm4v.Glm4vCausalLMOutputWithPast
或 tuple(torch.FloatTensor)
参数
- input_ids (形状为
(batch_size, sequence_length)
的torch.LongTensor
) — 词汇表中输入序列令牌的索引。默认情况下将忽略填充。索引可以使用 AutoTokenizer 获取。有关详情,请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。
- attention_mask (形状为
(batch_size, sequence_length)
的torch.Tensor
,可选) — 避免对填充令牌索引执行注意力的掩码。掩码值选择在[0, 1]
之间:- 1 表示未被掩盖的令牌,
- 0 表示被掩盖的令牌。
- position_ids (形状为
(batch_size, sequence_length)
的torch.LongTensor
,可选) — 每个输入序列令牌在位置嵌入中的位置索引。选择范围为[0, config.n_positions - 1]
。 - past_key_values (
list[torch.FloatTensor]
, 可选) — 预计算的隐藏状态(自注意力块和交叉注意力块中的键和值),可用于加速顺序解码。这通常包括模型在解码前期返回的past_key_values
,当use_cache=True
或config.use_cache=True
时。允许两种格式:
- 一个 Cache 实例,详见我们的 kv cache 指南;
- 一个长度为
config.n_layers
的tuple(torch.FloatTensor)
元组,每个元组包含两个形状为(batch_size, num_heads, sequence_length, embed_size_per_head)
的张量)。这也称为传统缓存格式。
模型将输出与输入相同的缓存格式。如果未传入
past_key_values
,则将返回传统缓存格式。如果使用
past_key_values
,用户可以选择只输入形状为(batch_size, 1)
的最后一个input_ids
(那些没有将过去键值状态提供给此模型的),而不是形状为(batch_size, sequence_length)
的所有input_ids
。 - inputs_embeds (形状为
(batch_size, sequence_length, hidden_size)
的torch.FloatTensor
,可选) — 可选地,您可以选择直接传递嵌入表示,而不是传递input_ids
。如果您想对如何将input_ids
索引转换为关联向量进行比模型内部嵌入查找矩阵更精细的控制,这会很有用。 - labels (形状为
(batch_size, sequence_length)
的torch.LongTensor
,可选) — 用于计算掩码语言模型损失的标签。索引应在[0, ..., config.vocab_size]
或 -100 之间(参见input_ids
文档字符串)。索引设置为-100
的令牌将被忽略(掩码),损失仅针对标签在[0, ..., config.vocab_size]
中的令牌计算。 - use_cache (
bool
, 可选) — 如果设置为True
,则返回past_key_values
键值状态,可用于加速解码(参见past_key_values
)。 - output_attentions (
bool
, 可选) — 是否返回所有注意力层的注意力张量。更多详情请参阅返回张量中的attentions
。 - output_hidden_states (
bool
, 可选) — 是否返回所有层的隐藏状态。更多详情请参阅返回张量中的hidden_states
。 - return_dict (
bool
, 可选) — 是否返回 ModelOutput 而不是普通的元组。 - pixel_values (形状为
(batch_size, num_channels, image_size, image_size)
的torch.Tensor
,可选) — 对应输入图像的张量。像素值可以使用{image_processor_class}
获取。有关详情,请参阅{image_processor_class}.__call__
({processor_class}
使用{image_processor_class}
处理图像)。 - pixel_values_videos (形状为 `(seq_length, num_channels * temporal_size * image_size * image_size)` 的
torch.FloatTensor
,可选) — 对应输入视频的张量。像素值可以使用 AutoImageProcessor 获取。有关详情,请参阅 Glm4vImageProcessor.call()。Glm4vProcessor 使用 Glm4vImageProcessor 处理视频。 - image_grid_thw (形状为
(num_images, 3)
的torch.LongTensor
,可选) — LLM 中每张图像的特征形状的时间、高度和宽度。 - video_grid_thw (形状为
(num_videos, 3)
的torch.LongTensor
,可选) — LLM 中每个视频的特征形状的时间、高度和宽度。 - rope_deltas (形状为
(batch_size, )
的torch.LongTensor
,可选) — 序列长度与多模态旋转位置编码之间的旋转位置编码索引差。 - cache_position (形状为
(sequence_length)
的torch.LongTensor
,可选) — 描述输入序列令牌在序列中位置的索引。与position_ids
不同,此张量不受填充影响。它用于在正确位置更新缓存并推断完整的序列长度。
返回
transformers.models.glm4v.modeling_glm4v.Glm4vCausalLMOutputWithPast
或 tuple(torch.FloatTensor)
一个 transformers.models.glm4v.modeling_glm4v.Glm4vCausalLMOutputWithPast
或一个 torch.FloatTensor
的元组(如果传入 return_dict=False
或当 config.return_dict=False
时),包含根据配置(Glm4vConfig)和输入的不同元素。
-
loss (
torch.FloatTensor
形状为(1,)
,可选,当提供labels
时返回) — 语言建模损失(用于下一个 token 预测)。 -
logits (形状为
(batch_size, sequence_length, config.vocab_size)
的torch.FloatTensor
) — 语言建模头部的预测分数(SoftMax 之前的每个词汇标记的分数)。 -
past_key_values (
tuple(tuple(torch.FloatTensor))
, 可选, 当传入use_cache=True
或当config.use_cache=True
时返回) — 长度为config.n_layers
的tuple(torch.FloatTensor)
元组,每个元组包含 2 个形状为(batch_size, num_heads, sequence_length, embed_size_per_head)
的张量。包含预计算的隐藏状态(自注意力块中的键和值),可用于(参见
past_key_values
输入)加速顺序解码。 -
hidden_states (
tuple[torch.FloatTensor]
, 可选, 当传入output_hidden_states=True
或当config.output_hidden_states=True
时返回) —torch.FloatTensor
的元组(如果模型有嵌入层,则一个用于嵌入层的输出,加上每个层的一个输出),形状为(batch_size, sequence_length, hidden_size)
。模型在每个层输出的隐藏状态以及可选的初始嵌入输出。
-
attentions (
tuple[torch.FloatTensor]
, 可选, 当传入output_attentions=True
或当config.output_attentions=True
时返回) —torch.FloatTensor
的元组(每个层一个),形状为(batch_size, num_heads, sequence_length, sequence_length)
。注意力 softmax 后的注意力权重,用于计算自注意力头中的加权平均值。
-
rope_deltas (形状为
(batch_size, )
的torch.LongTensor
,可选) — 序列长度与多模态旋转位置编码之间的旋转位置编码索引差。
Glm4vForConditionalGeneration 的前向方法,覆盖了 __call__
特殊方法。
尽管前向传播的配方需要在该函数中定义,但之后应该调用 Module
实例而不是此函数,因为前者负责运行预处理和后处理步骤,而后者则默默地忽略它们。
示例
>>> from PIL import Image
>>> import requests
>>> from transformers import AutoProcessor, Glm4vForConditionalGeneration
>>> model = Glm4vForConditionalGeneration.from_pretrained("THUDM/GLM-4.1V-9B-Thinking")
>>> processor = AutoProcessor.from_pretrained("THUDM/GLM-4.1V-9B-Thinking")
>>> messages = [
{
"role": "user",
"content": [
{"type": "image"},
{"type": "text", "text": "What is shown in this image?"},
],
},
]
>>> url = "https://www.ilankelman.org/stopsigns/australia.jpg"
>>> image = Image.open(requests.get(url, stream=True).raw)
>>> text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
>>> inputs = processor(text=[text], images=[image], vision_infos=[vision_infos])
>>> # Generate
>>> generate_ids = model.generate(inputs.input_ids, max_length=30)
>>> tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
"The image shows a street scene with a red stop sign in the foreground. In the background, there is a large red gate with Chinese characters ..."