add codes discussed in #73 that may help ipadapters

2024-02-07 18:57:57 -08:00
parent ef781cabcb
commit f63917a323
1 changed files with 3 additions and 1 deletions
@@ -388,7 +388,9 @@ class CrossAttentionPatch:
    def __call__(self, n, context_attn2, value_attn2, extra_options):
        org_dtype = n.dtype
        cond_or_uncond = extra_options["cond_or_uncond"]
-        sigma = extra_options["sigmas"][0].item() if 'sigmas' in extra_options else 999999999.9
+
+        sigma = extra_options["sigmas"][0] if 'sigmas' in extra_options else None
+        sigma = sigma.item() if sigma is not None else 999999999.9

        # extra options for AnimateDiff
        ad_params = extra_options['ad_params'] if "ad_params" in extra_options else None