Skip to content

Commit eeef1c4

Browse files
committed
chore: api change in modelopt 0.35
1 parent 806901c commit eeef1c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/llm/quantize_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515
except Exception:
1616
logger.warning("Unable to import quantization op. Please install modelopt library")
1717

18-
from modelopt.core.torch.quantization.qtensor.nvfp4_tensor import NVFP4QTensor
1918
from modelopt.torch.quantization.config import QuantizerAttributeConfig
2019
from modelopt.torch.quantization.nn.modules.tensor_quantizer import TensorQuantizer
20+
from modelopt.torch.quantization.qtensor.nvfp4_tensor import NVFP4QTensor
2121
from modelopt.torch.utils.dataset_utils import (
2222
create_forward_loop,
2323
get_dataset_dataloader,

0 commit comments

Comments
 (0)