Merge pull request #969 from modelscope/bugfix953

fix bug in issue 953
2025-09-30 13:00:15 +08:00
parent 7540568156 c63d474b60
commit 084bc2fc78
1 changed files with 1 additions and 1 deletions
--- a/diffsynth/vram_management/layers.py
+++ b/diffsynth/vram_management/layers.py
@@ -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