Make sure current_index exists, to match latest ComfyUI changes (backwards compatible)

2023-11-21 20:46:26 -06:00
parent ef7d6e883f
commit 36d72f3ae1
1 changed files with 2 additions and 1 deletions
@@ -100,7 +100,8 @@ def forward_timestep_embed(
            x = layer(x, context)
        elif isinstance(layer, SpatialTransformer):
            x = layer(x, context, transformer_options)
-            transformer_options["current_index"] += 1
+            if "current_index" in transformer_options:
+                transformer_options["current_index"] += 1
        elif isinstance(layer, openaimodel.Upsample):
            x = layer(x, output_shape=output_shape)
        else: