How to use vllm-sr/Vela-1.0-Encoder-307M-Modality with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="vllm-sr/Vela-1.0-Encoder-307M-Modality")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("vllm-sr/Vela-1.0-Encoder-307M-Modality") model = AutoModelForSequenceClassification.from_pretrained("vllm-sr/Vela-1.0-Encoder-307M-Modality", device_map="auto")
f4ee346 87db865 f4ee346 87db865 f4ee346
1
2
3
4
5
6
7
8
9
10
11
12
13
{ "label_to_idx": { "AR": 0, "BOTH": 2, "DIFFUSION": 1 }, "idx_to_label": { "0": "AR", "2": "BOTH", "1": "DIFFUSION" } }
