Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("Abigail45/Nina-Dolphin")
5model = AutoModelForCausalLM.from_pretrained(
6 "Abigail45/Nina-Dolphin",
7 device_map="auto",
8 load_in_4bit=True
9)
10
11prompt = "Summarize the causes of the French Revolution."
12inputs = tokenizer(prompt, return_tensors="pt")
13
14outputs = model.generate(
15 **inputs,
16 max_new_tokens=200,
17 do_sample=True,
18 temperature=0.7,
19 top_p=0.9
20)
21
22print(tokenizer.decode(outputs[0], skip_special_tokens=True))