From 6ad4177a75b553eada648fe5586de4703c34cb54 Mon Sep 17 00:00:00 2001 From: Qing Date: Sun, 4 Feb 2024 21:54:16 +0800 Subject: [PATCH] update diffusers==0.26.1, fix load single file sdxl inpainting model --- iopaint/tests/test_sd_model.py | 4 ++++ requirements.txt | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) 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