[Torch compile] Fix torch compile for controlnet (#4795)

Fix torch compile for controlnete
huggingface · Aug 28, 2023 · 5725b50 · 5725b50
1 parent e3380c0
commit 5725b50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/models/controlnet.py b/src/diffusers/models/controlnet.py
@@ -723,7 +723,7 @@ def forward(
             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)