fix bug in issue 953

2026-03-18 22:08:13 +00:00 · 2025-09-30 12:59:44 +08:00
parent 7540568156
commit c63d474b60
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