Views
No views yet
1from peft import AutoPeftModelForCausalLM
2from transformers import AutoTokenizer
3import torch
4
5model = AutoPeftModelForCausalLM.from_pretrained(
6 "your-username/echo-o1-nemotron-lora",
7 torch_dtype=torch.float16,
8 device_map="auto"
9)
10tokenizer = AutoTokenizer.from_pretrained("your-username/echo-o1-nemotron-lora")
11
12# Use Nemotron chat format
13prompt = """<extra_id_0>User</extra_id_0>
14Hi! I need help with my project.
15
16<extra_id_0>Assistant</extra_id_0>
17"""
18
19inputs = tokenizer(prompt, return_tensors="pt")
20outputs = model.generate(**inputs, max_new_tokens=200)
21response = tokenizer.decode(outputs[0], skip_special_tokens=True)
22Training Details
23
24Base Model: nvidia/Nemotron-Mini-4B-Instruct
25LoRA Rank: 8
26Target Modules: q_proj, v_proj
27Training Data: Conversational sales/consultation dataset