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

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

Fix torch compile for controlnete
parent fed88195
......@@ -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)
......
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