0xZohar commited on
Commit
005c492
·
verified ·
1 Parent(s): ab2c301

Fix: Use correct GPT model config (open_model.yaml)

Browse files

Changes config from open_model_v0.5.yaml to open_model.yaml to match the actual checkpoint architecture (23 layers, no embedder layers).

This resolves the AssertionError caused by config/checkpoint version mismatch.

Files changed (1) hide show
  1. code/demo.py +1 -1
code/demo.py CHANGED
@@ -57,7 +57,7 @@ def get_gpt_engine_cached():
57
  print("🔧 Initializing GPT engine (one-time setup)...")
58
 
59
  # Use absolute path relative to this file (fixes HF Spaces deployment)
60
- config_path = os.path.join(os.path.dirname(__file__), 'cube3d/configs/open_model_v0.5.yaml')
61
  gpt_ckpt_path = None # test mode doesn't use this
62
 
63
  # Detect HuggingFace Spaces environment
 
57
  print("🔧 Initializing GPT engine (one-time setup)...")
58
 
59
  # Use absolute path relative to this file (fixes HF Spaces deployment)
60
+ config_path = os.path.join(os.path.dirname(__file__), 'cube3d/configs/open_model.yaml')
61
  gpt_ckpt_path = None # test mode doesn't use this
62
 
63
  # Detect HuggingFace Spaces environment