fix clip guided (#1414)

patrickvonplaten · web-flow · commit f26cde3dff6b · 2022-11-25T12:04:40.000+01:00
diff --git a/examples/community/clip_guided_stable_diffusion.py b/examples/community/clip_guided_stable_diffusion.py
@@ -78,7 +78,8 @@ def __init__(
         )
 
         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.clip_model, False)

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,8 @@ def __init__(`
`78`	`78`	`)`
`79`	`79`
`80`	`80`	`self.normalize = transforms.Normalize(mean=feature_extractor.image_mean, std=feature_extractor.image_std)`
`81`		`- self.make_cutouts = MakeCutouts(feature_extractor.size)`
	`81`	`+ cut_out_size = feature_extractor.size if isinstance(feature_extractor.size, int) else feature_extractor.size["shortest_edge"]`
	`82`	`+ self.make_cutouts = MakeCutouts(cut_out_size)`
`82`	`83`
`83`	`84`	`set_requires_grad(self.text_encoder, False)`
`84`	`85`	`set_requires_grad(self.clip_model, False)`