jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

More code reuse in wan.

Browse Source 
Fix bug when changing the compute dtype on wan.
This commit is contained in:
comfyanonymous
2025-02-26 05:22:29 -05:00
parent 0844998db3
commit fa62287f1f
2 changed files with 6 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@@ -639,7 +639,7 @@ class ModelPatcher:
mem_counter += module_mem
load_completely.append((module_mem, n, m, params))
if cast_weight:
if cast_weight and hasattr(m, "comfy_cast_weights"):
m.prev_comfy_cast_weights = m.comfy_cast_weights
m.comfy_cast_weights = True