Merge pull request #969 from modelscope/bugfix953

fix bug in issue 953
This commit is contained in:
Zhongjie Duan
2025-09-30 13:00:15 +08:00
committed by GitHub

View File

@@ -116,7 +116,7 @@ class AutoWrappedLinear(torch.nn.Linear, AutoTorchModule):
self,
input: torch.Tensor,
weight: torch.Tensor,
bias: torch.Tensor | None = None,
bias: torch.Tensor = None,
) -> torch.Tensor:
device = input.device
origin_dtype = input.dtype