Muhammad Taqi Raza commited on
Commit
208621f
·
1 Parent(s): 47f4b64

out put video path

Browse files
Files changed (1) hide show
  1. cogvideo_controlnet_pcd.py +3 -0
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)