We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3670829 commit 1b5f7c0Copy full SHA for 1b5f7c0
tools/llm/quantize_utils.py
@@ -15,9 +15,9 @@
15
except Exception:
16
logger.warning("Unable to import quantization op. Please install modelopt library")
17
18
-from modelopt.core.torch.quantization.qtensor.nvfp4_tensor import NVFP4QTensor
19
from modelopt.torch.quantization.config import QuantizerAttributeConfig
20
from modelopt.torch.quantization.nn.modules.tensor_quantizer import TensorQuantizer
+from modelopt.torch.quantization.qtensor.nvfp4_tensor import NVFP4QTensor
21
from modelopt.torch.utils.dataset_utils import (
22
create_forward_loop,
23
get_dataset_dataloader,
0 commit comments