run ram 96gb rtx5060ti16gb ubuntu

#3
by zed122 - opened

(gsq) m@m-HP-Z440-Workstation:~/Desktop/GSQ-RCO-GGUF/GSQ$ vllm serve /home/m/Desktop/GSQ-RCO-GGUF/Llama-3.1-70B-Instruct-2Bit-GSQ
--cpu-offload-gb 20
--gpu-memory-utilization 0.90
--max-model-len 2048
--enforce-eager

#chat.py
from openai import OpenAI

client = OpenAI(
base_url="http://localhost:8000/v1",
api_key="none",
)

prompt = """<|begin_of_text|><|start_header_id|>user<|end_header_id|>

Hello! Introduce yourself briefly in two sentences.<|eot_id|><|start_header_id|>assistant<|end_header_id|>

"""

response = client.completions.create(
model="/home/m/Desktop/GSQ-RCO-GGUF/Llama-3.1-70B-Instruct-2Bit-GSQ",
prompt=prompt,
max_tokens=100,
temperature=0.6,
stop=["<|eot_id|>", "<|end_of_text|>"],
frequency_penalty=0.5,
)

print("Model Response:\n")
print(response.choices[0].text.strip())

Sign up or log in to comment