sd: load single file model using local config

2024-01-30 13:19:13 +08:00
parent 934b0fc455
commit e5f71ae535
12 changed files with 352 additions and 2 deletions
--- a/iopaint/tests/test_sd_model.py
+++ b/iopaint/tests/test_sd_model.py
@@ -232,9 +232,10 @@ def test_runway_sd_1_5_cpu_offload(device, strategy, sampler):
@pytest.mark.parametrize(
    "name",
    [
-        "sd-v1-5-inpainting.ckpt",
        "sd-v1-5-inpainting.safetensors",
        "v1-5-pruned-emaonly.safetensors",
+        "sd_xl_base_1.0.safetensors",
+        "sd_xl_base_1.0_inpainting_0.1.safetensors",
    ],
 )
 def test_local_file_path(device, sampler, name):