Views
No views yet
1from vllm import LLM, SamplingParams
2
3llm = LLM(
4 model="cococoomo/Exaone3.5-7.8B_ReST_V0_Quantized",
5 quantization="AWQ",
6)
7
8sampling_params = SamplingParams(
9 temperature=0.2,
10 top_p=0.8,
11 max_tokens=1024,
12)
13
14outputs = llm.generate(["Your prompt here"], sampling_params)
15print(outputs[0].outputs[0].text)