paddle12 commited on
Commit
0fe6652
·
verified ·
1 Parent(s): 3818144

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from utils import *
15
 
16
  # === [1] Model and Tokenizer Loading ===
17
  base_model_id = "NousResearch/Nous-Hermes-2-Mistral-7B-DPO"
18
- lora_path = "tat-llm-final-e3"
19
 
20
  # Load base model and LoRA adapter
21
  base_model = AutoModelForCausalLM.from_pretrained(base_model_id, torch_dtype=torch.float16)
 
15
 
16
  # === [1] Model and Tokenizer Loading ===
17
  base_model_id = "NousResearch/Nous-Hermes-2-Mistral-7B-DPO"
18
+ lora_path = "tat-llm-final-e4"
19
 
20
  # Load base model and LoRA adapter
21
  base_model = AutoModelForCausalLM.from_pretrained(base_model_id, torch_dtype=torch.float16)