Nano-Banana-PRO

Running on Zero

linoyts HF Staff commited on Aug 21

Commit

777d439

verified ·

1 Parent(s): f2e1d62

Update optimization.py

Files changed (1) hide show

optimization.py CHANGED Viewed

@@ -4,6 +4,7 @@
 from typing import Any
 from typing import Callable
 from typing import ParamSpec
 from torchao.quantization import Float8DynamicActivationFloat8WeightConfig
 import spaces
 import torch

 from typing import Any
 from typing import Callable
 from typing import ParamSpec
+from torchao.quantization import quantize_
 from torchao.quantization import Float8DynamicActivationFloat8WeightConfig
 import spaces
 import torch