LoganResearch commited on
Commit
b66d66c
·
1 Parent(s): 26ec3cc

Fix MODEL_PATH to use merged model

Browse files
Files changed (1) hide show
  1. ubermenschetien_v2_full.py +1 -1
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 = "NousResearch/Hermes-3-Llama-3.1-8B"
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")