wan2-2-fp8da-aoti-faster

Running on Zero

linoyts HF Staff commited on Aug 1

Commit

7019232

verified ·

1 Parent(s): ff5ea08

Update optimization.py

Files changed (1) hide show

optimization.py CHANGED Viewed

@@ -15,6 +15,7 @@ from torchao.quantization import Int8WeightOnlyConfig
 from optimization_utils import capture_component_call
 from optimization_utils import aoti_compile
 from optimization_utils import ZeroGPUCompiledModel
 P = ParamSpec('P')

 from optimization_utils import capture_component_call
 from optimization_utils import aoti_compile
 from optimization_utils import ZeroGPUCompiledModel
+from optimization_utils import drain_module_parameters
 P = ParamSpec('P')