wrong fix

9b2c5b6b · Wq-dd · a6dff8f4 · 9b2c5b6b
Commit 9b2c5b6b authored Jul 21, 2025 by Wq-dd
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lightx2v/common/ops/mm/mm_weight.py lightx2v/common/ops/mm/mm_weight.py +1 -1

No files found.
--- a/lightx2v/common/ops/mm/mm_weight.py
+++ b/lightx2v/common/ops/mm/mm_weight.py
@@ -26,7 +26,7 @@ except ImportError:
    deep_gemm = None

 try:
-    from torch.ao.quantization.utils import quant_int8_per_token_matmul, quantize_activation_per_token_absmax
+    from torchao.quantization.utils import quant_int8_per_token_matmul, quantize_activation_per_token_absmax
 except ModuleNotFoundError:
    quant_int8_per_token_matmul, quantize_activation_per_token_absmax = None, None