From 5d4413001b7ced46c96aace85454bd3525d2f3d5 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 24 Aug 2023 10:19:47 +0000 Subject: [PATCH] make style --- .../pipelines/stable_diffusion/convert_from_ckpt.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index eec04df2c9..0a534d4193 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -1387,7 +1387,7 @@ def download_from_original_stable_diffusion_ckpt( unet = UNet2DConditionModel(**unet_config) if is_accelerate_available(): - if model_type not in ["SDXL", "SDXL-Refiner"]: # SBM Delay this. + if model_type not in ["SDXL", "SDXL-Refiner"]: # SBM Delay this. for param_name, param in converted_unet_checkpoint.items(): set_module_tensor_to_device(unet, param_name, "cpu", value=param) else: @@ -1589,7 +1589,7 @@ def download_from_original_stable_diffusion_ckpt( checkpoint, config_name, prefix="conditioner.embedders.1.model.", has_projection=True, **config_kwargs ) - if is_accelerate_available(): # SBM Now move model to cpu. + if is_accelerate_available(): # SBM Now move model to cpu. if model_type in ["SDXL", "SDXL-Refiner"]: for param_name, param in converted_unet_checkpoint.items(): set_module_tensor_to_device(unet, param_name, "cpu", value=param) @@ -1617,11 +1617,11 @@ def download_from_original_stable_diffusion_ckpt( checkpoint, config_name, prefix="conditioner.embedders.0.model.", has_projection=True, **config_kwargs ) - if is_accelerate_available(): # SBM Now move model to cpu. + if is_accelerate_available(): # SBM Now move model to cpu. if model_type in ["SDXL", "SDXL-Refiner"]: for param_name, param in converted_unet_checkpoint.items(): set_module_tensor_to_device(unet, param_name, "cpu", value=param) - + pipe = StableDiffusionXLImg2ImgPipeline( vae=vae, text_encoder=text_encoder,