alternative refiner implementation

2023-08-06 22:08:52 +03:00
parent 57e8a11d17
commit 6c7b6ecb81
6 changed files with 100 additions and 13 deletions
@@ -305,7 +305,7 @@ class KDiffusionSampler:
        shared.total_tqdm.update()

    def launch_sampling(self, steps, func):
-        state.sampling_steps = steps
+        state.sampling_steps = self.stop_at if self.stop_at is not None else steps
        state.sampling_step = 0

        try: