参数

• images (ImageInput) — 要预处理的图像。 期望单个或批量的图像，像素值范围为 0 到 255。 如果传入像素值在 0 到 1 之间的图像，请设置 do_rescale=False。
• do_pad (bool, 可选, 默认为 self.do_pad) — 是否根据最长边将图像填充为正方形。 填充值由 image_mean 参数确定。
• do_resize (bool, 可选, 默认为 self.do_resize) — 是否调整图像大小。
• size (Dict[str, int], 可选, 默认为 self.size) — 调整大小后图像的尺寸。图像的最短边将调整为 size[“shortest_edge”]，最长边将调整以保持输入宽高比。
• resample (int, 可选, 默认为 self.resample) — 如果调整图像大小，则使用的重采样过滤器。可以是枚举 PILImageResampling 中的一个。仅在 do_resize 设置为 True 时有效。
• do_center_crop (bool, 可选, 默认为 self.do_center_crop) — 是否对图像进行中心裁剪。
• crop_size (Dict[str, int], 可选, 默认为 self.crop_size) — 中心裁剪的尺寸。仅在 do_center_crop 设置为 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 时有效。
• do_convert_rgb (bool, 可选, 默认为 self.do_convert_rgb) — 是否将图像转换为 RGB 格式。
• return_tensors (str 或 TensorType, 可选) — 返回张量的类型。可以是以下之一：
  • Unset: 返回 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)。
  • Unset: 使用输入图像的通道维度格式。
• 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)。

参数

• images (ImageInput) — 要预处理的图像。 接受单个或批量的图像，像素值范围为 0 到 255。如果传入的图像像素值介于 0 和 1 之间，请设置 do_rescale=False。
• do_resize (bool, 可选, 默认为 self.do_resize) — 是否调整图像大小。
• size (Dict[str, int], 可选, 默认为 self.size) — 描述模型的最大输入尺寸。
• resample (PILImageResampling 或 InterpolationMode, 可选, 默认为 self.resample) — 调整图像大小时使用的重采样滤波器。可以是枚举类型 PILImageResampling 之一。 仅当 do_resize 设置为 True 时才有效。
• do_center_crop (bool, 可选, 默认为 self.do_center_crop) — 是否对图像进行中心裁剪。
• crop_size (Dict[str, int], 可选, 默认为 self.crop_size) — 应用 center_crop 后输出图像的大小。
• 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 时才有效。
• do_convert_rgb (bool, 可选, 默认为 self.do_convert_rgb) — 是否将图像转换为 RGB 格式。
• return_tensors (str 或 TensorType, 可选, 默认为 self.return_tensors) — 如果设置为 `pt`，则返回堆叠张量，否则返回张量列表。
• data_format (ChannelDimension 或 str, 可选, 默认为 self.data_format) — 仅支持 ChannelDimension.FIRST。为了与慢速处理器兼容而添加。
• input_data_format (ChannelDimension 或 str, 可选, 默认为 self.input_data_format) — 输入图像的通道维度格式。如果未设置，则通道维度格式从输入图像推断。可以是以下之一：
  • "channels_first" 或 ChannelDimension.FIRST: 图像格式为 (num_channels, height, width)。
  • "channels_last" 或 ChannelDimension.LAST: 图像格式为 (height, width, num_channels)。
  • "none" 或 ChannelDimension.NONE: 图像格式为 (height, width)。
• device (torch.device, 可选, 默认为 self.device) — 处理图像的设备。如果未设置，则设备从输入图像推断。 do_pad (bool, 可选, 默认为 self.do_pad): 是否基于最长边将图像填充为正方形。可以被 do_pad 参数覆盖

参数

• input_ids (torch.LongTensor，形状为 (batch_size, sequence_length)) — 词汇表中输入序列 token 的索引。 如果您提供 padding，默认情况下 padding 将被忽略。

  索引可以使用 AutoTokenizer 获取。 有关详细信息，请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。

  什么是输入 ID？

• pixel_values (torch.FloatTensor，形状为 (batch_size, num_channels, image_size, image_size)) -- 对应于输入图像的 tensor。 像素值可以使用 [AutoImageProcessor](/docs/transformers/v4.50.0/en/model_doc/auto#transformers.AutoImageProcessor) 获取。 有关详细信息，请参阅 [CLIPImageProcessor.__call__()](/docs/transformers/v4.50.0/en/model_doc/vilt#transformers.ViltFeatureExtractor.__call__) (`LlavaProcessor` 使用 CLIPImageProcessor 处理图像)。
• attention_mask (torch.Tensor，形状为 (batch_size, sequence_length), 可选) — 用于避免在 padding token 索引上执行 attention 的 Mask。 Mask 值在 [0, 1] 中选择：
  • 1 表示 token 未被 mask，
  • 0 表示 token 已被 mask。

  什么是 attention mask？

  索引可以使用 AutoTokenizer 获取。 有关详细信息，请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。

  如果使用 past_key_values，则可以选择仅输入最后的 decoder_input_ids（请参阅 past_key_values）。

  如果您想更改 padding 行为，您应该阅读 modeling_opt._prepare_decoder_attention_mask 并根据您的需要进行修改。 有关默认策略的更多信息，请参见 论文 中的图 1。

  • 1 表示 head 未被 mask，
  • 0 表示 head 已被 mask。
• position_ids (torch.LongTensor，形状为 (batch_size, sequence_length), 可选) — 位置嵌入中每个输入序列 token 的位置索引。 在范围 [0, config.n_positions - 1] 中选择。 什么是位置 ID？
• past_key_values (tuple(tuple(torch.FloatTensor)), 可选, 当传递 use_cache=True 或当 config.use_cache=True 时返回) — 长度为 config.n_layers 的 tuple(tuple(torch.FloatTensor))，其中每个 tuple 都有 2 个形状为 (batch_size, num_heads, sequence_length, embed_size_per_head) 的 tensor 和 2 个形状为 (batch_size, num_heads, encoder_sequence_length, embed_size_per_head) 的附加 tensor。

  包含预先计算的 hidden-state（self-attention 块和 cross-attention 块中的 key 和 value），可以用于（请参阅 past_key_values 输入）加速顺序解码。

  如果使用 past_key_values，则用户可以选择仅输入最后的 decoder_input_ids（那些没有将其过去的 key value 状态提供给此模型的），形状为 (batch_size, 1)，而不是所有形状为 (batch_size, sequence_length) 的 decoder_input_ids。

• inputs_embeds (torch.FloatTensor，形状为 (batch_size, sequence_length, hidden_size), 可选) — 或者，您可以选择直接传递嵌入表示，而不是传递 input_ids。 如果您希望比模型的内部嵌入查找矩阵更精细地控制如何将 input_ids 索引转换为关联向量，这将非常有用。
• vision_feature_layer (Union[int, List[int]], 可选, 默认为 -2) — 用于选择视觉特征的层的索引。 如果提供多个索引，则会将相应索引的视觉特征连接起来以形成视觉特征。
• vision_feature_select_strategy (str, 可选, 默认为 "default") — 用于从视觉 backbone 中选择视觉特征的特征选择策略。 可以是 "default" 或 "full" 之一。
• use_cache (bool, 可选) — 如果设置为 True，则返回 past_key_values 键值状态，并且可以用于加速解码（请参阅 past_key_values）。
• output_attentions (bool, 可选) — 是否返回所有 attention 层的 attention tensor。 有关更多详细信息，请参见返回的 tensor 下的 attentions。
• output_hidden_states (bool, 可选) — 是否返回所有层的 hidden state。 有关更多详细信息，请参见返回的 tensor 下的 hidden_states。
• return_dict (bool, 可选) — 是否返回 ModelOutput 而不是纯 tuple。
• cache_position (torch.LongTensor，形状为 (sequence_length), 可选) — 索引，描述输入序列 token 在序列中的位置。 与 position_ids 相反，此 tensor 不受 padding 的影响。 它用于在正确的位置更新缓存并推断完整的序列长度。
• labels (torch.LongTensor，形状为 (batch_size, sequence_length), 可选) — 用于计算 masked language modeling loss 的标签。 索引应为 [0, ..., config.vocab_size] 或 -100（请参阅 input_ids 文档字符串）。 索引设置为 -100 的 token 将被忽略（masked），loss 仅针对标签在 [0, ..., config.vocab_size] 中的 token 计算。
• logits_to_keep (int 或 torch.Tensor, 可选) — 如果是 int，则计算最后 logits_to_keep 个 token 的 logits。 如果是 0，则计算所有 input_ids 的 logits（特殊情况）。 只需要最后一个 token 的 logits 用于生成，并且仅针对该 token 计算可以节省内存，这对于长序列或大型词汇表大小而言变得非常重要。 如果是 torch.Tensor，则必须是与序列长度维度中要保留的索引相对应的 1D 张量。 这在使用 packed tensor 格式（批次和序列长度的单维度）时很有用。

一个 transformers.models.llava.modeling_llava.LlavaCausalLMOutputWithPast 或一个 torch.FloatTensor 元组 (如果传递了 return_dict=False 或当 config.return_dict=False 时)，其中包含各种元素，具体取决于配置 (LlavaConfig) 和输入。

• loss (torch.FloatTensor，形状为 (1,)，可选，当提供 labels 时返回) — 语言建模损失（用于下一个 token 预测）。

• logits (torch.FloatTensor，形状为 (batch_size, sequence_length, config.vocab_size)) — 语言建模头的预测分数（SoftMax 之前每个词汇表 token 的分数）。

• 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 之后的注意力权重，用于计算自注意力头中的加权平均值。

• image_hidden_states (torch.FloatTensor, 可选) — 大小为 (batch_size, num_images, sequence_length, hidden_size)` 的 torch.FloatTensor。 视觉编码器生成并在投影最后一个隐藏状态之后的模型的 image_hidden_states。