Merge pull request #9 from blepping/fix_default_image_only_indicator

Fix crash on accessing default_image_only_indicator
2024-03-05 07:36:38 -08:00
parent 93e2bc1842 cb715de8ab
commit 3ce68d4a12
1 changed files with 1 additions and 1 deletions
@@ -156,7 +156,7 @@ def __temp__forward(self, x, timesteps=None, context=None, y=None, control=None,
        transformer_patches = transformer_options.get("patches", {})

        num_video_frames = kwargs.get("num_video_frames", self.default_num_video_frames)
-        image_only_indicator = kwargs.get("image_only_indicator", self.default_image_only_indicator)
+        image_only_indicator = kwargs.get("image_only_indicator", getattr(self, "default_image_only_indicator", None))
        time_context = kwargs.get("time_context", None)

        assert (y is not None) == (