Views
No views yet
1python -m vllm.entrypoints.openai.api_server \
2 --model neutrino12/tensorstax-sft-unsloth-32b-dora32-3020 \
3 --host 0.0.0.0 \
4 --port 8000 \
5 --max-model-len 32768 \
6 --trust-remote-code1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "neutrino12/tensorstax-sft-unsloth-32b-dora32-3020",
5 torch_dtype="auto",
6 device_map="auto",
7 trust_remote_code=True
8)
9
10tokenizer = AutoTokenizer.from_pretrained(
11 "neutrino12/tensorstax-sft-unsloth-32b-dora32-3020",
12 trust_remote_code=True
13)
14
15# Generate text
16inputs = tokenizer("Your prompt here", return_tensors="pt")
17outputs = model.generate(**inputs, max_new_tokens=512)
18response = tokenizer.decode(outputs[0], skip_special_tokens=True)