torch_frame.config

文本嵌入器

TextEmbedderConfig

文本嵌入模型,将字符串/句子列表映射为PyTorch张量嵌入。

TextTokenizerConfig

文本分词器,将字符串/句子列表映射到MultiNestedTensor的字典中。

ModelConfig

可学习的模型,将单列的TensorData对象映射为行嵌入。

ImageEmbedderConfig

图像嵌入模型,将图像列表映射为PyTorch张量嵌入。

ImageEmbedder

image_embedder 的父类,用于 ImageEmbedderConfig