Update nodes_sampler.py

2025-10-01 19:30:44 +03:00
parent 1316f57390
commit 4b079bdf5a
1 changed files with 4 additions and 3 deletions
@@ -1034,9 +1034,10 @@ class WanVideoSampler:
                        is_uncond=True
                    )
                    log.info(f"Extracted {len(lynx_ref_buffer_uncond)} uncond ref buffers")
-
-                lynx_embeds["ip_x"] = lynx_embeds["ip_x"].to(device, dtype)
-                lynx_embeds["ip_x_uncond"] = lynx_embeds["ip_x_uncond"].to(device, dtype)
+                
+                if lynx_embeds.get("ip_x", None) is not None:
+                    lynx_embeds["ip_x"] = lynx_embeds["ip_x"].to(device, dtype)
+                    lynx_embeds["ip_x_uncond"] = lynx_embeds["ip_x_uncond"].to(device, dtype)
                lynx_embeds["ref_feature_extractor"] = False
                lynx_embeds["ref_latent"] = lynx_embeds["ref_text_embed"] = None
                lynx_embeds["ref_buffer"] = lynx_ref_buffer