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

fix wan vram bug

Browse Source
This commit is contained in:
Artiprocher
2025-11-18 23:17:15 +08:00
parent f85af085df
commit 82378a2815
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
diffsynth/core/vram/layers.py
View File

@@ -255,8 +255,7 @@ class AutoWrappedNonRecurseModule(AutoWrappedModule):
getattr(self, name).to("meta") getattr(self, name).to("meta")
def cast_to(self, module, dtype, device): def cast_to(self, module, dtype, device):
for name in self.required_params: # Parameter casting is implemented in the model architecture.
getattr(module, name).to(dtype=dtype, device=device)
return module return module
def __getattr__(self, name): def __getattr__(self, name):