Update fine_tune_inference_test_mistral.py
Browse files
fine_tune_inference_test_mistral.py
CHANGED
|
@@ -14,7 +14,7 @@ os.environ["TORCH_HOME"] = "/app/.torch_cache"
|
|
| 14 |
os.makedirs("/app/.torch_cache", exist_ok=True)
|
| 15 |
|
| 16 |
# === Ayarlar
|
| 17 |
-
MODEL_BASE = "mistralai/Mistral-7B-Instruct-v0.2"
|
| 18 |
USE_FINE_TUNE = False
|
| 19 |
FINE_TUNE_REPO = "UcsTurkey/trained-zips"
|
| 20 |
FINE_TUNE_ZIP = "trained_model_000_009.zip"
|
|
|
|
| 14 |
os.makedirs("/app/.torch_cache", exist_ok=True)
|
| 15 |
|
| 16 |
# === Ayarlar
|
| 17 |
+
MODEL_BASE = "malhajar/Mistral-7B-Instruct-v0.2-turkish" #"mistralai/Mistral-7B-Instruct-v0.2"
|
| 18 |
USE_FINE_TUNE = False
|
| 19 |
FINE_TUNE_REPO = "UcsTurkey/trained-zips"
|
| 20 |
FINE_TUNE_ZIP = "trained_model_000_009.zip"
|