Spaces:
Running
on
Zero
Running
on
Zero
ACE-Step Custom
commited on
Commit
·
5b76ce1
1
Parent(s):
140e1e2
Try _fast_init=False to prevent meta device initialization
Browse files- acestep/handler.py +3 -2
acestep/handler.py
CHANGED
|
@@ -492,8 +492,9 @@ class AceStepHandler:
|
|
| 492 |
acestep_v15_checkpoint_path,
|
| 493 |
trust_remote_code=True,
|
| 494 |
attn_implementation=candidate,
|
| 495 |
-
|
| 496 |
-
low_cpu_mem_usage=False, #
|
|
|
|
| 497 |
)
|
| 498 |
attn_implementation = candidate
|
| 499 |
break
|
|
|
|
| 492 |
acestep_v15_checkpoint_path,
|
| 493 |
trust_remote_code=True,
|
| 494 |
attn_implementation=candidate,
|
| 495 |
+
torch_dtype=self.dtype,
|
| 496 |
+
low_cpu_mem_usage=False, # Disable memory-efficient weight loading
|
| 497 |
+
_fast_init=False, # Disable fast initialization (prevents meta device)
|
| 498 |
)
|
| 499 |
attn_implementation = candidate
|
| 500 |
break
|