Update models.py

kijai · Dec 3, 2024 · fa45b71 · fa45b71
1 parent 73b49a3
commit fa45b71
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hyvideo/modules/models.py b/hyvideo/modules/models.py
@@ -597,7 +597,8 @@ def forward(
         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: