Update models.py

This commit is contained in:
kijai
2024-12-03 19:14:00 +02:00
parent 73b49a30a6
commit fa45b717fe
+2 -1
View File
@@ -597,7 +597,8 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin):
vec = self.time_in(t) vec = self.time_in(t)
# text modulation # text modulation
vec = vec + self.vector_in(text_states_2) if text_states_2 is not None:
vec = vec + self.vector_in(text_states_2)
# guidance modulation # guidance modulation
if self.guidance_embed: if self.guidance_embed: