Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
| 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)
|