Hugging Face's logo

Spaces:
zerogpu-aoti
/
wan2-2-fp8da-aoti
Running on Zero

App Files Community
4
linoyts HF Staff commited on
Commit
b09f6a2
·
verified ·
1 Parent(s): d3f7631

Update optimization.py

Browse files
Files changed (1) hide show
  1. optimization.py +1 -1
optimization.py CHANGED
@@ -15,7 +15,7 @@ from torchao.quantization import Int8WeightOnlyConfig
15
  from optimization_utils import capture_component_call
16
  from optimization_utils import aoti_compile
17
  from optimization_utils import ZeroGPUCompiledModel
18
-
19
 
20
  P = ParamSpec('P')
21
 
 
15
  from optimization_utils import capture_component_call
16
  from optimization_utils import aoti_compile
17
  from optimization_utils import ZeroGPUCompiledModel
18
+ from optimization_utils import drain_module_parameters
19
 
20
  P = ParamSpec('P')
21