torch_frame.config — pytorch-frame documentation (original) (raw)
Contents
Text Embedder
| TextEmbedderConfig | Text embedder model that maps a list of strings/sentences into PyTorch Tensor embeddings. |
|---|---|
| TextTokenizerConfig | Text tokenizer that maps a list of strings/sentences into a dictionary of MultiNestedTensor. |
| ModelConfig | Learnable model that maps a single-column TensorData object into row embeddings. |
| ImageEmbedderConfig | Image embedder model that maps a list of images into PyTorch Tensor embeddings. |
| ImageEmbedder | Parent class for the image_embedder of ImageEmbedderConfig. |