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