fix

lshqqytiger · Feb 12, 2024 · 835ee20 · 835ee20 · f1am3d · Feb 16, 2024
1 parent c7fc860
commit 835ee20
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/modules/onnx_impl/pipelines/onnx_stable_diffusion_xl_img2img_pipeline.py b/modules/onnx_impl/pipelines/onnx_stable_diffusion_xl_img2img_pipeline.py
@@ -29,7 +29,6 @@ def __init__(
     ):
         optimum.onnxruntime.ORTStableDiffusionXLImg2ImgPipeline.__init__(self, vae_decoder, text_encoder, unet, config, tokenizer, scheduler, feature_extractor, vae_encoder, text_encoder_2, tokenizer_2, use_io_binding, model_save_dir, add_watermarker)
         super().__init__()
-        del self.image_processor # This image processor requires np array. In order to share same workflow with non-XL pipelines, delete it.
 
     def prepare_latents(self, image, timestep, batch_size, num_images_per_prompt, dtype, generator=None):
         batch_size = batch_size * num_images_per_prompt

diff --git a/modules/onnx_impl/pipelines/onnx_stable_diffusion_xl_pipeline.py b/modules/onnx_impl/pipelines/onnx_stable_diffusion_xl_pipeline.py
@@ -27,7 +27,6 @@ def __init__(
     ):
         optimum.onnxruntime.ORTStableDiffusionXLPipeline.__init__(self, vae_decoder, text_encoder, unet, config, tokenizer, scheduler, feature_extractor, vae_encoder, text_encoder_2, tokenizer_2, use_io_binding, model_save_dir, add_watermarker)
         super().__init__()
-        del self.image_processor # This image processor requires np array. In order to share same workflow with non-XL pipelines, delete it.
 
     def prepare_latents(self, batch_size, num_channels_latents, height, width, dtype, generator, latents=None):
         return prepare_latents(self.scheduler.init_noise_sigma, batch_size, height, width, dtype, generator, latents, num_channels_latents, self.vae_scale_factor)