Commit
·
b66d66c
1
Parent(s):
26ec3cc
Fix MODEL_PATH to use merged model
Browse files
ubermenschetien_v2_full.py
CHANGED
|
@@ -61,7 +61,7 @@ LOGS_DIR = os.path.join(ROOT, "improvement_logs")
|
|
| 61 |
ROLLBACK_DIR = os.path.join(ROOT, "rollback_checkpoints")
|
| 62 |
|
| 63 |
# Model paths
|
| 64 |
-
MODEL_PATH = "
|
| 65 |
DENSE_CHECKPOINT = os.path.join(ROOT, "dense_checkpoints/step_100")
|
| 66 |
CFHOT_CHECKPOINT = os.path.join(ROOT, "cfhot_checkpoints/ckpt_5000")
|
| 67 |
MULTI_HEAD_DIR = os.path.join(ROOT, "multi_head_checkpoints")
|
|
|
|
| 61 |
ROLLBACK_DIR = os.path.join(ROOT, "rollback_checkpoints")
|
| 62 |
|
| 63 |
# Model paths
|
| 64 |
+
MODEL_PATH = "LoganResearch/ARC-Base-8B-Condensed"
|
| 65 |
DENSE_CHECKPOINT = os.path.join(ROOT, "dense_checkpoints/step_100")
|
| 66 |
CFHOT_CHECKPOINT = os.path.join(ROOT, "cfhot_checkpoints/ckpt_5000")
|
| 67 |
MULTI_HEAD_DIR = os.path.join(ROOT, "multi_head_checkpoints")
|