ACE-Step Custom commited on
Commit
897bbd6
·
1 Parent(s): 2538e84

Use official ACE-Step demo approach: dtype as string (bfloat16)

Browse files
Files changed (1) hide show
  1. acestep/handler.py +1 -2
acestep/handler.py CHANGED
@@ -497,8 +497,7 @@ class AceStepHandler:
497
  acestep_v15_checkpoint_path,
498
  trust_remote_code=True,
499
  attn_implementation=candidate,
500
- torch_dtype=self.dtype,
501
- device_map={"": self.device}, # Force all components to target device immediately
502
  )
503
  attn_implementation = candidate
504
  break
 
497
  acestep_v15_checkpoint_path,
498
  trust_remote_code=True,
499
  attn_implementation=candidate,
500
+ dtype="bfloat16" # Use string like official demo
 
501
  )
502
  attn_implementation = candidate
503
  break