ACE-Step Custom commited on
Commit
5b76ce1
·
1 Parent(s): 140e1e2

Try _fast_init=False to prevent meta device initialization

Browse files
Files changed (1) hide show
  1. 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
- dtype=self.dtype,
496
- low_cpu_mem_usage=False, # Force immediate weight loading
 
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