nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'pr/799'

Browse Source
This commit is contained in:
kijai
2025-07-17 19:33:10 +03:00
parent 6e5b493b28 88f50a8fd9
commit e011d031a8
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils.py
+5 -2
View File
@@ -73,7 +73,10 @@ def apply_lora(model, device_to, transformer_load_device, params_to_keep=None, d
set_module_tensor_to_device(model.model.diffusion_model, key, device=transformer_load_device, dtype=dtype_to_use, value=state_dict[key])
except:
continue
model.patch_weight_to_device("{}.{}".format(name, param), device_to=device_to)
if low_mem_load:
model.patch_weight_to_device("{}.{}".format(name, param), device_to=device_to, inplace_update=True)
else:
model.patch_weight_to_device("{}.{}".format(name, param), device_to=device_to)
if low_mem_load:
try:
set_module_tensor_to_device(model.model.diffusion_model, key, device=transformer_load_device, dtype=dtype_to_use, value=model.model.diffusion_model.state_dict()[key])
@@ -237,4 +240,4 @@ def fourier_filter(x, scale_low=1.0, scale_high=1.5, freq_cutoff=20):
def is_image_black(image, threshold=1e-3):
if image.min() < 0:
image = (image + 1) / 2
return torch.all(image < threshold).item()
return torch.all(image < threshold).item()