ACE-Step Custom commited on
Commit
6a590ee
·
1 Parent(s): fa7f63d

Fix DiT initialization: pass model name only, not full path

Browse files
Files changed (1) hide show
  1. src/ace_step_engine.py +3 -1
src/ace_step_engine.py CHANGED
@@ -110,9 +110,11 @@ class ACEStepEngine:
110
  logger.info(f"Initializing DiT handler with model: {dit_model_path}")
111
 
112
  # Initialize DiT handler (handles main diffusion model, VAE, text encoder)
 
 
113
  status_dit, success_dit = self.dit_handler.initialize_service(
114
  project_root=project_root,
115
- config_path=str(checkpoints_dir / dit_model_path),
116
  device="auto",
117
  use_flash_attention=False,
118
  compile_model=False,
 
110
  logger.info(f"Initializing DiT handler with model: {dit_model_path}")
111
 
112
  # Initialize DiT handler (handles main diffusion model, VAE, text encoder)
113
+ # Note: handler auto-detects checkpoints dir as project_root/checkpoints
114
+ # config_path should be just the model name, not full path
115
  status_dit, success_dit = self.dit_handler.initialize_service(
116
  project_root=project_root,
117
+ config_path=dit_model_path, # Just model name, handler adds checkpoints/
118
  device="auto",
119
  use_flash_attention=False,
120
  compile_model=False,