From 7f43f65235085b976188e6edb519d4cd79b6a509 Mon Sep 17 00:00:00 2001 From: Will Berman Date: Sun, 26 Feb 2023 23:03:23 -0800 Subject: [PATCH] image_noiser -> image_normalizer comment (#2496) --- .../stable_diffusion/pipeline_stable_unclip_img2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py index 8fdc47a786..b17df264a6 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py @@ -169,7 +169,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline): device = torch.device(f"cuda:{gpu_id}") - # TODO: self.image_noiser.{scale,unscale} are not covered by the offload hooks, so they fails if added to the list + # TODO: self.image_normalizer.{scale,unscale} are not covered by the offload hooks, so they fails if added to the list models = [ self.image_encoder, self.text_encoder,