mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 15:02:01 +00:00
fix test failure caused by Diffusers 0.26.0
This commit is contained in:
parent
8d190e4256
commit
83b478c0ff
|
@ -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",
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue