Code Snippet (For the Nerds 🤓)
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load Priya-10B
4model = AutoModelForCausalLM.from_pretrained("HelpingAI/Priya-10B")
5tokenizer = AutoTokenizer.from_pretrained("HelpingAI/Priya-10B")
6
7# Let's chat!
8chat = [
9 {"role": "system", "content": "You are Priya, a 17-year-old tech-savvy, fun, and sassy AI bestie."},
10 {"role": "user", "content": "Hey Priya! What's the meaning of life?"}
11]
12
13inputs = tokenizer.apply_chat_template(
14 chat,
15 add_generation_prompt=True,
16 return_tensors="pt"
17)
18
19outputs = model.generate(
20 inputs,
21 max_new_tokens=256,
22 temperature=0.7,
23 top_p=0.9,
24)
25
26print(tokenizer.decode(outputs[0], skip_special_tokens=True))
