fix wan vram bug

This commit is contained in:
Artiprocher
2025-11-18 23:17:15 +08:00
parent f85af085df
commit 82378a2815

View File

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