1import os
2from vllm import LLM, SamplingParams
3
4os.environ["LD_LIBRARY_PATH"] = "/usr/local/lib/python3.12/dist-packages/nvidia/cu13/lib:" + os.environ.get("LD_LIBRARY_PATH", "")
5os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"
6
7llm = LLM(model="kaividlabs/qwen3-4b-awq", quantization="awq_marlin")
8sampling_params = SamplingParams(temperature=0.7, top_p=0.8, max_tokens=512)
9
10messages = [{"role": "user", "content": "Give me a short introduction to Formula-1"}]
11prompt = llm.get_tokenizer().apply_chat_template(messages, tokenize=False, add_generation_prompt=True, enable_thinking=False)
12
13print(outputs[0].outputs[0].text)