"...git@developer.sourcefind.cn:OpenDAS/diffusers.git" did not exist on "1d2587bb34f55c1362c44c4734dae0e138f466ac"
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): ...@@ -723,7 +723,7 @@ class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlnetMixin):
class_emb = self.class_embedding(class_labels).to(dtype=self.dtype) class_emb = self.class_embedding(class_labels).to(dtype=self.dtype)
emb = emb + class_emb 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": if self.config.addition_embed_type == "text":
aug_emb = self.add_embedding(encoder_hidden_states) 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