Update models.py
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user