Update models.py

2024-12-03 19:14:00 +02:00
parent 73b49a30a6
commit fa45b717fe
1 changed files with 2 additions and 1 deletions
@@ -597,7 +597,8 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin):
        vec = self.time_in(t)

        # 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
        if self.guidance_embed: