fix test failure caused by Diffusers 0.26.0

This commit is contained in:
Pierre Chapuis 2024-02-04 07:55:31 +01:00
parent 8d190e4256
commit 83b478c0ff
2 changed files with 3 additions and 2 deletions

View file

@ -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",

View file

@ -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)