slarkprime/Llama-2-7b-QLoRA

app.py

@@ -17,7 +17,8 @@ login(hf_token)
 dataset = load_dataset("json", data_files="data.jsonl")
 
 # Definisci il modello e il tokenizer
-model_name = "meta-llama/Llama-2-7b-hf"
+#model_name = "meta-llama/Llama-2-7b-hf"
+model_name = "slarkprime/Llama-2-7b-QLoRA"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)