point Space at v17b model_final.pt
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ from transformers import AutoTokenizer
|
|
| 20 |
HUB_REPO = os.environ.get("HUB_REPO", "ProjectScugnizz/scugnizz-llama-pcs")
|
| 21 |
HUB_PATH = os.environ.get(
|
| 22 |
"HUB_PATH",
|
| 23 |
-
"training-runs/sft-chat-
|
| 24 |
)
|
| 25 |
SCRIPT_REPO = os.environ.get("SCRIPT_REPO", "ProjectScugnizz/scugnizz-llama-training")
|
| 26 |
MODEL_MOUNT = Path(os.environ.get("MODEL_MOUNT", "/models"))
|
|
@@ -50,8 +50,8 @@ _state = {
|
|
| 50 |
"status": "cold",
|
| 51 |
"hub_path": HUB_PATH.strip("/"),
|
| 52 |
# ponytail: Space var WEIGHTS_FILE so mid-run pulse can ship without waiting for final
|
| 53 |
-
"weights_file": (os.environ.get("WEIGHTS_FILE") or "
|
| 54 |
-
or "
|
| 55 |
}
|
| 56 |
|
| 57 |
|
|
|
|
| 20 |
HUB_REPO = os.environ.get("HUB_REPO", "ProjectScugnizz/scugnizz-llama-pcs")
|
| 21 |
HUB_PATH = os.environ.get(
|
| 22 |
"HUB_PATH",
|
| 23 |
+
"training-runs/sft-chat-v17b-tool-answer-rl-8b-20260816-220712",
|
| 24 |
)
|
| 25 |
SCRIPT_REPO = os.environ.get("SCRIPT_REPO", "ProjectScugnizz/scugnizz-llama-training")
|
| 26 |
MODEL_MOUNT = Path(os.environ.get("MODEL_MOUNT", "/models"))
|
|
|
|
| 50 |
"status": "cold",
|
| 51 |
"hub_path": HUB_PATH.strip("/"),
|
| 52 |
# ponytail: Space var WEIGHTS_FILE so mid-run pulse can ship without waiting for final
|
| 53 |
+
"weights_file": (os.environ.get("WEIGHTS_FILE") or "model_final.pt").strip()
|
| 54 |
+
or "model_final.pt",
|
| 55 |
}
|
| 56 |
|
| 57 |
|