Unverified Commit 7f43f652 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

image_noiser -> image_normalizer comment (#2496)

parent 6960e722
...@@ -169,7 +169,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline): ...@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment