andreabac3 commited on
Commit
ef030f7
·
1 Parent(s): 2b6f242

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Fauno - Italian LLM
2
+
3
+
4
+ ## Model initialization
5
+ ```python
6
+ from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
7
+ from peft import PeftModel
8
+
9
+ tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-7b-hf")
10
+ model = LlamaForCausalLM.from_pretrained(
11
+ "decapoda-research/llama-7b-hf",
12
+ load_in_8bit=True,
13
+ device_map="auto",
14
+ )
15
+ model = PeftModel.from_pretrained(model, "andreabac3/Fauno-Italian-LLM-7B")
16
+ model.eval()
17
+ ```
18
+
19
+ ## Inference
20
+ ```python
21
+ def evaluate(question: str) -> str:
22
+ prompt = f"The conversation between human and AI assistant.\n[|Human|] {question}.\n[|AI|] "
23
+ inputs = tokenizer(prompt, return_tensors="pt")
24
+ input_ids = inputs["input_ids"].cuda()
25
+ generation_output = model.generate(
26
+ input_ids=input_ids,
27
+ generation_config=generation_config,
28
+ return_dict_in_generate=True,
29
+ output_scores=True,
30
+ max_new_tokens=256
31
+ )
32
+ output = tokenizer.decode(generation_output.sequences[0]).split("[|AI|]")[1]
33
+ return output
34
+
35
+ your_question: str = "Qual'è il significato della vita?"
36
+ print(evaluate(your_question))
37
+ ```
38
+
39
+ ### Output
40
+ ```
41
+ Il senso della vita è una domanda che molte persone hanno cercato di rispondere per secoli.
42
+ Alla fine, il senso della vita è soggettivo e varia da persona a persona.
43
+ Alcune persone credono che il senso della vita sia trovare la felicità, mentre altre credono che sia raggiungere i propri obiettivi o aiutare gli altri.
44
+ Alla fine, il senso della vita è determinato dall'individuo e dai loro valori e credenze.
45
+ In definitiva, il senso della vita è qualcosa che ognuno deve trovare da solo.
46
+ Non c'è una risposta giusta o sbagliata, poiché ogni persona ha le proprie convinzioni e credenze.
47
+ La ricerca del senso della vita può essere un viaggio lungo e difficile, ma vale la pena perseguire.
48
+ ```