diff --git a/iopaint/tests/test_sd_model.py b/iopaint/tests/test_sd_model.py index 6865e5a..aa26c71 100644 --- a/iopaint/tests/test_sd_model.py +++ b/iopaint/tests/test_sd_model.py @@ -256,10 +256,14 @@ def test_local_file_path(device, sampler, name): name = f"device_{device}_{sampler}_{name}" + is_sdxl = "sd_xl" in name + assert_equal( model, cfg, f"sd_local_model_{name}.png", img_p=current_dir / "overture-creations-5sI6fQgYIuo.png", mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", + fx=1.5 if is_sdxl else 1, + fy=1.5 if is_sdxl else 1, ) diff --git a/requirements.txt b/requirements.txt index 10b6634..f8aea20 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ torch>=2.0.0 opencv-python -diffusers==0.25.1 +diffusers==0.26.1 accelerate peft==0.7.1 transformers>=4.35.1