UniPC: use same default for solver_type

Fixes a bug when switching from UniPC from another scheduler (i.e., DEIS) that uses a different solver type. The solver is now the same as if we had instantiated the scheduler directly.
2025-12-06 20:44:33 +08:00 · 2023-07-04 14:46:55 +02:00
1 changed files with 1 additions and 1 deletions
--- a/src/diffusers/schedulers/scheduling_unipc_multistep.py
+++ b/src/diffusers/schedulers/scheduling_unipc_multistep.py
@@ -173,7 +173,7 @@ class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin):

        if solver_type not in ["bh1", "bh2"]:
            if solver_type in ["midpoint", "heun", "logrho"]:
-                self.register_to_config(solver_type="bh1")
+                self.register_to_config(solver_type="bh2")
            else:
                raise NotImplementedError(f"{solver_type} does is not implemented for {self.__class__}")