From 480a4b4ae42aecef198a9a30fdc793e059b565fa Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Fri, 19 Jan 2024 07:34:49 +0000 Subject: [PATCH] update --- src/diffusers/loaders/single_file.py | 1 - src/diffusers/loaders/single_file_utils.py | 3 +-- .../pipeline_stable_diffusion_xl_img2img.py | 1 + 3 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py index d07b58bac5..1e4678cfbe 100644 --- a/src/diffusers/loaders/single_file.py +++ b/src/diffusers/loaders/single_file.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -import inspect import os import re diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index 3ab0aa716f..8002f1d9ef 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -215,9 +215,8 @@ def infer_original_config_file(class_name, checkpoint): def fetch_original_config(pipeline_class_name, checkpoint, original_config_file=None): - def is_valid_url(url): - pattern = r'^(http|https):\/\/([\w.-]+)(\.[\w.-]+)+([\/\w\.-]*)*\/?$' + pattern = r"^(http|https):\/\/([\w.-]+)(\.[\w.-]+)+([\/\w\.-]*)*\/?$" return bool(re.match(pattern, url)) if os.path.isfile(original_config_file): diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py index 4e95a876ce..1c22affba1 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py @@ -792,6 +792,7 @@ class StableDiffusionXLImg2ImgPipeline( self.unet.config.addition_time_embed_dim * len(add_time_ids) + text_encoder_projection_dim ) expected_add_embed_dim = self.unet.add_embedding.linear_1.in_features + if ( expected_add_embed_dim > passed_add_embed_dim and (expected_add_embed_dim - passed_add_embed_dim) == self.unet.config.addition_time_embed_dim