Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix test failure caused by Diffusers 0.26.0

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pyproject.toml
View file

@ -27,7 +27,7 @@ training = [
"tomli>=2.0.1", "tomli>=2.0.1",
] ]
test = [ test = [
"diffusers>=0.24.0", "diffusers>=0.26.1",
"transformers>=4.35.2", "transformers>=4.35.2",
"piq>=0.8.0", "piq>=0.8.0",
"invisible-watermark>=0.2.0", "invisible-watermark>=0.2.0",
@ -37,7 +37,7 @@ test = [
"segment-anything-py>=1.0", "segment-anything-py>=1.0",
] ]
conversion = [ conversion = [
"diffusers>=0.24.0", "diffusers>=0.26.1",
"transformers>=4.35.2", "transformers>=4.35.2",
"segment-anything-py>=1.0", "segment-anything-py>=1.0",
"requests>=2.26.0", "requests>=2.26.0",

1
tests/foundationals/latent_diffusion/test_solvers.py
View file

@ -30,6 +30,7 @@ def test_dpm_solver_diffusers(n_steps: int, last_step_first_order: bool):
beta_end=0.012, beta_end=0.012,
lower_order_final=False, lower_order_final=False,
euler_at_final=last_step_first_order, euler_at_final=last_step_first_order,
final_sigmas_type="sigma_min", # default before Diffusers 0.26.0
) )
diffusers_scheduler.set_timesteps(n_steps) diffusers_scheduler.set_timesteps(n_steps)
refiners_scheduler = DPMSolver(num_inference_steps=n_steps, last_step_first_order=last_step_first_order) refiners_scheduler = DPMSolver(num_inference_steps=n_steps, last_step_first_order=last_step_first_order)