From 0b621e61c3fd84e76ea31e555bb380427baa4d71 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sat, 26 Aug 2023 22:30:02 +0200 Subject: [PATCH] [Torch compile] Fix torch compile for controlnet (#4795) Fix torch compile for controlnete --- src/diffusers/models/controlnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/controlnet.py b/src/diffusers/models/controlnet.py index ed3f3e6871..4bf3886a99 100644 --- a/src/diffusers/models/controlnet.py +++ b/src/diffusers/models/controlnet.py @@ -723,7 +723,7 @@ class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlnetMixin): class_emb = self.class_embedding(class_labels).to(dtype=self.dtype) emb = emb + class_emb - if "addition_embed_type" in self.config: + if self.config.addition_embed_type is not None: if self.config.addition_embed_type == "text": aug_emb = self.add_embedding(encoder_hidden_states)