kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

image_noiser -> image_normalizer comment (#2496)

Browse Source
This commit is contained in:
Will Berman
2023-02-26 23:03:23 -08:00
committed by GitHub
parent 6960e72225
commit 7f43f65235
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py
+1 -1
View File
@@ -169,7 +169,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline):
device = torch.device(f"cuda:{gpu_id}") 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 = [ models = [
self.image_encoder, self.image_encoder,
self.text_encoder, self.text_encoder,