Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from peft import PeftModel
|
|
| 5 |
|
| 6 |
# --- Configuration ---
|
| 7 |
# 1. Base Model ID: Llama-2-7b-chat-hf is typically used as the base
|
| 8 |
-
base_model_id = "
|
| 9 |
|
| 10 |
# 2. LoRA Path: IMPORTANT! Replace this with the path to your fine-tuned model
|
| 11 |
# This should be the Hugging Face repo ID (e.g., "your-username/llama2-dockerfile-lora")
|
|
|
|
| 5 |
|
| 6 |
# --- Configuration ---
|
| 7 |
# 1. Base Model ID: Llama-2-7b-chat-hf is typically used as the base
|
| 8 |
+
base_model_id = "NousResearch/Llama-2-7b-chat-hf"
|
| 9 |
|
| 10 |
# 2. LoRA Path: IMPORTANT! Replace this with the path to your fine-tuned model
|
| 11 |
# This should be the Hugging Face repo ID (e.g., "your-username/llama2-dockerfile-lora")
|