From b079f0b2c87398a8f7c4a8ede17d8d9628f55147 Mon Sep 17 00:00:00 2001 From: Qiacheng Date: Sun, 20 Aug 2023 22:49:06 -0700 Subject: [PATCH] fix refiner pipe for sdxl, added invisible-watermark in requirements --- requirements.txt | 1 + requirements_versions.txt | 2 +- scripts/openvino_accelerate.py | 4 ++-- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 00d00ab5..2b9c4fc0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -33,3 +33,4 @@ torchsde transformers==4.30.0 diffusers==0.20.0 openvino==2023.1.0.dev20230811 +invisible-watermark \ No newline at end of file diff --git a/requirements_versions.txt b/requirements_versions.txt index 3f397e00..e67008c8 100644 --- a/requirements_versions.txt +++ b/requirements_versions.txt @@ -31,4 +31,4 @@ torchsde==0.2.5 transformers==4.30.0 diffusers==0.20.0 openvino==2023.1.0.dev20230811 - +invisible-watermark diff --git a/scripts/openvino_accelerate.py b/scripts/openvino_accelerate.py index 59b861b5..7e39a7b2 100644 --- a/scripts/openvino_accelerate.py +++ b/scripts/openvino_accelerate.py @@ -337,9 +337,9 @@ def get_diffusers_sd_model(model_config, vae_config, sampler_name, enable_cachin else: sd_model = StableDiffusionXLPipeline.from_single_file(checkpoint_path, local_config_file=checkpoint_config, load_safety_checker=False, torch_dtype=torch.float32) if (mode == 1): - sd_model = StableDiffusionXLImg2ImgPipeline(**sd_model.components) + sd_model = StableDiffusionXLImg2ImgPipeline.from_single_file(checkpoint_path, local_config_file=checkpoint_config, load_safety_checker=False, torch_dtype=torch.float32) elif (mode == 2): - sd_model = StableDiffusionXLInpaintPipeline(**sd_model.components) + sd_model = StableDiffusionXLInpaintPipeline.from_single_file(checkpoint_path, local_config_file=checkpoint_config, load_safety_checker=False, torch_dtype=torch.float32) else: if model_config != "None": local_config_file = os.path.join(curr_dir_path, 'configs', model_config)