Tiny fix wrong comments (#9589)
This commit is contained in:
@@ -876,7 +876,6 @@ class ModelOptNvFp4FusedMoEMethod(FusedMoEMethodBase):
|
||||
data=torch.empty(
|
||||
layer.num_local_experts,
|
||||
2 * intermediate_size_per_partition,
|
||||
# 2 fp4 items are packed in the input dimension
|
||||
hidden_size // self.quant_config.group_size,
|
||||
dtype=weight_scale_dtype,
|
||||
),
|
||||
@@ -895,7 +894,6 @@ class ModelOptNvFp4FusedMoEMethod(FusedMoEMethodBase):
|
||||
data=torch.empty(
|
||||
layer.num_local_experts,
|
||||
hidden_size,
|
||||
# 2 fp4 items are packed in the input dimension
|
||||
intermediate_size_per_partition // self.quant_config.group_size,
|
||||
dtype=weight_scale_dtype,
|
||||
),
|
||||
|
||||
Reference in New Issue
Block a user