Muhammad Taqi Raza
commited on
Commit
·
208621f
1
Parent(s):
47f4b64
out put video path
Browse files
cogvideo_controlnet_pcd.py
CHANGED
@@ -172,6 +172,9 @@ class CogVideoXControlnetPCD(ModelMixin, ConfigMixin, PeftAdapterMixin):
|
|
172 |
timestep_cond: Optional[torch.Tensor] = None,
|
173 |
return_dict: bool = True,
|
174 |
):
|
|
|
|
|
|
|
175 |
hidden_states = torch.cat([hidden_states, controlnet_states], dim=2)
|
176 |
|
177 |
# controlnet_states = self.controlnext_encoder(controlnet_states, timestep=timestep)
|
|
|
172 |
timestep_cond: Optional[torch.Tensor] = None,
|
173 |
return_dict: bool = True,
|
174 |
):
|
175 |
+
print("hidden_states.shape =", hidden_states.shape)
|
176 |
+
print("controlnet_states.shape =", controlnet_states.shape)
|
177 |
+
|
178 |
hidden_states = torch.cat([hidden_states, controlnet_states], dim=2)
|
179 |
|
180 |
# controlnet_states = self.controlnext_encoder(controlnet_states, timestep=timestep)
|