Update README.md

README.md

@@ -26,20 +26,17 @@ That paper got accepted with the scores 5, 4, 4 out of 5
 ## How to run the model
 ```python
 from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM
-from peft import PeftModel, PeftConfig
 device = "cuda" # the device to load the model onto
 
-peft_model_id = "andreabac3/DanteLLM_instruct_7b-v0.2-boosted"
-config = PeftConfig.from_pretrained(peft_model_id)
-model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, device_map="auto", load_in_8bit=True)
-model = PeftModel.from_pretrained(model, peft_model_id)
+model_id = "rstless-research/DanteLLM-7B-Instruct-Italian-v0.1"
+model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", load_in_8bit=True)
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 
 model.eval()
 
 messages = [
     {"role": "user", "content": "Ciao chi sei?"},
-    {"role": "assistant", "content": "Ciao, sono Open Fauno, un large language model. Come posso aiutarti?"},
+    {"role": "assistant", "content": "Ciao, sono DanteLLM, un large language model. Come posso aiutarti?"},
     {"role": "user", "content": "Quanto dista la Terra dalla Luna?"}
 ]