Skip to content

Commit 196748e

Browse files
committed
address reviews
Signed-off-by: Zhiyu Cheng <zhiyuc@nvidia.com>
1 parent ea44272 commit 196748e

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

modelopt/torch/export/moe_utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@ def _export_fused_experts(module: nn.Module, dtype: torch.dtype) -> None:
4343
{E}.up_proj.weight, {E}.up_proj.weight_scale, ...
4444
{E}.down_proj.weight, {E}.down_proj.weight_scale, ...
4545
"""
46-
from modelopt.torch.export.layer_utils import set_expert_quantizer_amax
4746
from modelopt.torch.export.unified_export_hf import _export_quantized_weight
4847
from modelopt.torch.quantization.plugins.huggingface import _get_fused_expert_intermediate_dim
4948

0 commit comments

Comments
 (0)