diff --git a/nodes_sampler.py b/nodes_sampler.py index d6064ae..e1cbfc8 100644 --- a/nodes_sampler.py +++ b/nodes_sampler.py @@ -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