Views
No views yet
1"Di seguito è riportata un'istruzione che descrive un compito, abbinata a un input che fornisce un ulteriore contesto. Scrivete una risposta che completi in modo appropriato la richiesta.
2
3### Istruzione:
4{instruction}
5
6### Input:
7{input}
8
9### Risposta:"1import os
2import torch
3import sys
4from transformers import LlamaTokenizer, LlamaForCausalLM
5
6if torch.cuda.is_available():
7 device = "cuda"
8else:
9 device = "cpu"
10
11def generate_prompt(instruction, input=None):
12 if input:
13 return f"""Di seguito è riportata un'istruzione che descrive un compito, abbinata a un input che fornisce un ulteriore contesto. Scrivete una risposta che completi in modo appropriato la richiesta.
14
15### Istruzione:
16{instruction}
17
18### Input:
19{input}
20
21### Risposta:"""
22 else:
23 return f"""Di seguito è riportata un'istruzione che descrive un compito. Scrivete una risposta che completi in modo appropriato la richiesta..
24
25### Istruzione:
26{instruction}
27
28### Risposta:"""
29
30model_name = "raicrits/OpenLLama13b_Loquace_ITA"
31
32model = LlamaForCausalLM.from_pretrained(
33 model_name,
34 torch_dtype=torch.float16,
35 device_map="auto"
36)
37
38tokenizer = LlamaTokenizer.from_pretrained(model_name)
39
40instruction = "qual'è la relazione tra i seguenti oggetti"
41input = "sedia, tavolo, divano"
42prompt = generate_prompt("instruction", input)
43inputs = tokenizer(prompt, return_tensors="pt")
44input_ids = inputs["input_ids"].to(device)
45
46generation_output = model.generate(
47 input_ids=input_ids,
48 max_new_tokens=256,
49)
50
51
52output = tokenizer.decode(generation_output[0])
53output = output.split("### Risposta:")[1].strip().replace("</s>","")
54print(output)"Sedia, tavolo e divano sono tutti oggetti che possono essere utilizzati per creare un'atmosfera rilassante in una stanza."