Skip to content

GitLab

Commit 9b2c5b6b authored by Wq-dd's avatar Wq-dd
Browse files

wrong fix

parent a6dff8f4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/common/ops/mm/mm_weight.py
View file @ 9b2c5b6b
...@@ -26,7 +26,7 @@ except ImportError: ...@@ -26,7 +26,7 @@ except ImportError:
deep_gemm = None deep_gemm = None
try: 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: except ModuleNotFoundError:
quant_int8_per_token_matmul, quantize_activation_per_token_absmax = None, None quant_int8_per_token_matmul, quantize_activation_per_token_absmax = None, None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment