properly reset caches etc on t=0

2025-02-27 16:33:39 +01:00
parent 1629a5c16d
commit b2466a6454
1 changed files with 5 additions and 0 deletions
@@ -94,6 +94,11 @@ class RASManager:
            self.timestep = self.timestep_from_sigmas(
                transformer_options["sigmas"], transformer_options["sample_sigmas"]
            )
+            if self.timestep == 0:
+                # reset as much as possible
+                self.live_img_indices = None
+                self.live_txt_indices = None
+                self.drop_count = None
            return forward_orig(*args, **kwargs)

        model.forward_orig = MethodType(new_forward, model)