Fix modelopt FP8 create weights (#18447)

Signed-off-by: root <dafrimi@nvidia.com>
This commit is contained in:
danielafrimi
2026-02-17 00:59:50 +08:00
committed by GitHub
parent eba6af385d
commit 0ff24159a5
2 changed files with 13 additions and 4 deletions
@@ -461,6 +461,8 @@ class ModelOptFp8LinearMethod(LinearMethodBase):
layer: torch.nn.Module,
input_size_per_partition: int,
output_partition_sizes: List[int],
input_size: Optional[int],
output_size: Optional[int],
params_dtype: torch.dtype,
**extra_weight_attrs,
) -> None: