Update nodes.py

This commit is contained in:
kijai 2024-10-13 23:05:50 +03:00
parent 42c46b76e1
commit 0f5d67a9a9

View File

@ -836,6 +836,7 @@ class CogVideoImageEncode:
if not pipeline["cpu_offloading"]: if not pipeline["cpu_offloading"]:
vae.to(device) vae.to(device)
check_diffusers_version()
vae._clear_fake_context_parallel_cache() vae._clear_fake_context_parallel_cache()
input_image = image.clone() input_image = image.clone()