image_noiser -> image_normalizer comment (huggingface#2496)

CCRcmcpe · Feb 27, 2023 · 7f43f65 · 7f43f65
1 parent 6960e72
commit 7f43f65
Showing 1 changed file with 1 addition and 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
@@ -169,7 +169,7 @@ def enable_sequential_cpu_offload(self, gpu_id=0):
 
         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,