Update README.md

README.md

@@ -29,13 +29,23 @@ We evaluated the model using the same test sets as used for the Open Ita LLM Lea
 
 ## Usage
 
+Be sure to have transformers, peft and sentencepiece installed
+
+```python
+pip install transformers peft sentencepiece
+```
+
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from peft import PeftModel, PeftConfig
 
 device = "cuda"
 
-model = AutoModelForCausalLM.from_pretrained("MoxoffSpA/xxxx")
-tokenizer = AutoTokenizer.from_pretrained("MoxoffSpA/xxxx")
+config = PeftConfig.from_pretrained("MoxoffSpA/xxxx")
+model = AutoModelForCausalLM.from_pretrained("alpindale/Mistral-7B-v0.2-hf")
+tokenizer = AutoTokenizer.from_pretrained("alpindale/Mistral-7B-v0.2-hf")
+
+model = PeftModel.from_pretrained(model, "MoxoffSpA/xxxx")
 
 messages = [
     {"role": "user", "content": "Qual è il tuo piatto preferito??"},