Spaces:
Running
on
Zero
Running
on
Zero
ACE-Step Custom
commited on
Commit
·
6a590ee
1
Parent(s):
fa7f63d
Fix DiT initialization: pass model name only, not full path
Browse files- 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=
|
| 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,
|