Views
No views yet

1from transformers import AutoTokenizer
2from gptqmodel import GPTQModel
3
4tokenizer = AutoTokenizer.from_pretrained("ModelCloud/DeepSeek-R1-Distill-Qwen-7B-gptqmodel-4bit-vortex-v1")
5model = GPTQModel.load("ModelCloud/DeepSeek-R1-Distill-Qwen-7B-gptqmodel-4bit-vortex-v1")
6
7messages = [
8 {"role": "system", "content": "You are a helpful and harmless assistant. You should think step-by-step."},
9 {"role": "user", "content": "How can I design a data structure in C++ to store the top 5 largest integer numbers?"},
10]
11input_tensor = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt")
12
13outputs = model.generate(input_ids=input_tensor.to(model.device), max_new_tokens=512)
14result = tokenizer.decode(outputs[0][input_tensor.shape[1]:], skip_special_tokens=True)
15
16print(result)