fix timestepping on windows?

2024-02-13 17:30:34 +01:00
parent d87b7ce1c1
commit 6a411dcb2c
1 changed files with 2 additions and 1 deletions
@@ -454,7 +454,8 @@ 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