Skip to content

GitLab

Unverified Commit f26cde3d authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix clip guided (#1414)

parent 9f10c545
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/community/clip_guided_stable_diffusion.py
View file @ f26cde3d
...@@ -78,7 +78,8 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline): ...@@ -78,7 +78,8 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline):
) )
self.normalize = transforms.Normalize(mean=feature_extractor.image_mean, std=feature_extractor.image_std) self.normalize = transforms.Normalize(mean=feature_extractor.image_mean, std=feature_extractor.image_std)
self.make_cutouts = MakeCutouts(feature_extractor.size) cut_out_size = feature_extractor.size if isinstance(feature_extractor.size, int) else feature_extractor.size["shortest_edge"]
self.make_cutouts = MakeCutouts(cut_out_size)
set_requires_grad(self.text_encoder, False) set_requires_grad(self.text_encoder, False)
set_requires_grad(self.clip_model, False) set_requires_grad(self.clip_model, False)
......
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