fix test failure caused by Diffusers 0.26.0

catwell · catwell · commit 83b478c0ff51 · 2024-02-04T08:58:22.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,7 @@ training = [
     "tomli>=2.0.1",
 ]
 test = [
-    "diffusers>=0.24.0",
+    "diffusers>=0.26.1",
     "transformers>=4.35.2",
     "piq>=0.8.0",
     "invisible-watermark>=0.2.0",
@@ -37,7 +37,7 @@ test = [
     "segment-anything-py>=1.0",
 ]
 conversion = [
-    "diffusers>=0.24.0",
+    "diffusers>=0.26.1",
     "transformers>=4.35.2",
     "segment-anything-py>=1.0",
     "requests>=2.26.0",
diff --git a/tests/foundationals/latent_diffusion/test_solvers.py b/tests/foundationals/latent_diffusion/test_solvers.py
@@ -30,6 +30,7 @@ def test_dpm_solver_diffusers(n_steps: int, last_step_first_order: bool):
         beta_end=0.012,
         lower_order_final=False,
         euler_at_final=last_step_first_order,
+        final_sigmas_type="sigma_min",  # default before Diffusers 0.26.0
     )
     diffusers_scheduler.set_timesteps(n_steps)
     refiners_scheduler = DPMSolver(num_inference_steps=n_steps, last_step_first_order=last_step_first_order)

Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,7 @@ def test_dpm_solver_diffusers(n_steps: int, last_step_first_order: bool):`
`30`	`30`	`beta_end=0.012,`
`31`	`31`	`lower_order_final=False,`
`32`	`32`	`euler_at_final=last_step_first_order,`
	`33`	`+ final_sigmas_type="sigma_min", # default before Diffusers 0.26.0`
`33`	`34`	`)`
`34`	`35`	`diffusers_scheduler.set_timesteps(n_steps)`
`35`	`36`	`refiners_scheduler = DPMSolver(num_inference_steps=n_steps, last_step_first_order=last_step_first_order)`