Spaces:
Running on Zero
Running on Zero
ACE-Step Custom commited on
Commit ·
897bbd6
1
Parent(s): 2538e84
Use official ACE-Step demo approach: dtype as string (bfloat16)
Browse files- 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 |
-
|
| 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
|