Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from peft import PeftModel
|
|
| 4 |
import torch
|
| 5 |
|
| 6 |
# Directory where your fine-tuned Phi-2 model and associated files are stored.
|
| 7 |
-
model_dir = "./phi2-
|
| 8 |
|
| 9 |
# Directory to store offloaded model parts (for large models).
|
| 10 |
offload_dir = "./offload"
|
|
|
|
| 4 |
import torch
|
| 5 |
|
| 6 |
# Directory where your fine-tuned Phi-2 model and associated files are stored.
|
| 7 |
+
model_dir = "./phi2-qlora-finetuned"
|
| 8 |
|
| 9 |
# Directory to store offloaded model parts (for large models).
|
| 10 |
offload_dir = "./offload"
|