Views
No views yet
cosimoiaia/Loquace-7B-Mistral.1from transformers import MistralForCausalLM, AutoTokenizer
2
3
4def generate_prompt(instruction):
5 prompt = f"""### Instruction: {instruction}
6
7### Response:
8"""
9 return prompt
10
11model_name = "cosimoiaia/Loquace-7B-Mistral"
12
13model = MistralForCausalLM.from_pretrained(
14 model_name,
15 device_map="auto",
16 torch_dtype=torch.bfloat16
17)
18
19model.config.use_cache = True
20
21
22tokenizer = AutoTokenizer.from_pretrained(model_name, add_eos_token=False)
23
24prompt = generate_prompt("Chi era Dante Alighieri?")
25inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
26
27outputs = model.generate(**inputs, do_sample = True, num_beams = 2, top_k=50, top_p= 0.95, max_new_tokens=2046, early_stopping = True)
28print(tokenizer.decode(outputs[0], skip_special_tokens=True).split("Response:")[1].strip())> Pretendi di essere Dante Alighieri e raccontami la tua vita.
Sono Dante Alighieri, un poeta e scrittore italiano del XIV secolo. Nacqui a Firenze nel 1265 e ho trascorso gran parte della mia vita viaggiando in Europa e scrivendo poesie. Ho anche scritto la Divina Commedia, una delle opere letterarie più famose di tutti i tempi.
> Quanto fa 7 x 14 - 98 ?
7 x 14 = 98, quindi la risposta è 0.
> La radice quadrata di 144 ?
La radice quadrata di 144 è 12.
> Inventa una ricetta pugliese con la polenta.
La ricetta è chiamata "Polenta al burro di arachidi e formaggio". Per prepararla, inizia bollendo la polenta secondo le istruzioni sulla confezione. Mentre la polenta si cucina, mescola insieme 1/2 tazza di burro di arachidi, 1/4 di tazza di formaggio grattugiato e 1/4 di tazza di parmigiano grattugiato. Una volta che la polenta è cotta, versala su una teglia da forno e distribuire il composto di burro di arachidi e formaggio sopra la polenta. Metti in forno a 350 gradi per 15 minuti o fino a quando la parte superiore è dorata. Servi caldo con un'insalata di pomodoro e basilico fresco.