theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-18 22:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #969 from modelscope/bugfix953

Browse Source 
fix bug in issue 953
This commit is contained in:
Zhongjie Duan
2025-09-30 13:00:15 +08:00
committed by GitHub
parent 7540568156 c63d474b60
commit 084bc2fc78
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/vram_management/layers.py
View File

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