Update kugelaudio_open/schedule/dpm_solver.py

kugelaudio_open/schedule/dpm_solver.py

@@ -294,6 +294,9 @@ class DPMSolverMultistepScheduler(SchedulerMixin, ConfigMixin):
         self._begin_index = None
         if not self.sigmas.is_meta:
             self.sigmas = self.sigmas.to("cpu")  # to avoid too much CPU/GPU communication
+
+    @property
+    def step_index(self):
         """
         The index counter for current timestep. It will increase 1 after each scheduler step.
         """
@@ -420,6 +423,8 @@ class DPMSolverMultistepScheduler(SchedulerMixin, ConfigMixin):
         self._begin_index = None
         if not self.sigmas.is_meta:
             self.sigmas = self.sigmas.to("cpu")  # to avoid too much CPU/GPU communication
+
+    # Copied from diffusers.schedulers.scheduling_ddpm.DDPMScheduler._threshold_sample
     def _threshold_sample(self, sample: torch.Tensor) -> torch.Tensor:
         """
         "Dynamic thresholding: At each sampling step we set s to a certain percentile absolute pixel value in xt0 (the