ACE-Step-Training / acestep /handler.py

Commit History

Fix meta tensor device mismatch by also patching torch.arange
f266123
unverified

pedroapfilho commited on

Patch both residual_fsq and finite_scalar_quantization tensor refs
011d30d
unverified

pedroapfilho commited on

Fix meta tensor: patch residual_fsq module's tensor reference, not torch.tensor
0c47532
unverified

pedroapfilho commited on

Fix meta tensor: monkey-patch ResidualFSQ to force CPU tensor creation
bf4c82d
unverified

pedroapfilho commited on

Fix meta tensor crash: force CPU device context during model init
b0a0560
unverified

pedroapfilho commited on

Try _fast_init=False to prevent meta device initialization
5b76ce1

ACE-Step Custom commited on

Fix dtype parameter: use dtype instead of torch_dtype, remove device_map
140e1e2

ACE-Step Custom commited on

Fix meta tensor: add device_map=None and use torch_dtype parameter
a11c404

ACE-Step Custom commited on

Fix meta tensor error: disable low_cpu_mem_usage for model loading
23dc982

ACE-Step Custom commited on

Deploy ACE-Step Custom Edition with bug fixes
a602628

ACE-Step Custom commited on