xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-28 17:17:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove prints

Browse Source
This commit is contained in:
kijai 2024-10-02 20:27:38 +03:00
parent f9f06d595e
commit 4fb602cad5
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cogvideox_fun/pipeline_cogvideox_control.py
View File

@ -689,9 +689,7 @@ class CogVideoX_Fun_Pipeline_Control(VideoSysPipeline):
) )
latents_tile = latents[:, input_start_t:input_end_t,:, :, :] latents_tile = latents[:, input_start_t:input_end_t,:, :, :]
print("latents_tile.shape", latents_tile.shape)
control_latents_tile = control_latents[:, input_start_t:input_end_t, :, :, :] control_latents_tile = control_latents[:, input_start_t:input_end_t, :, :, :]
print("control_latents_tile.shape", control_latents_tile.shape)
latent_model_input_tile = torch.cat([latents_tile] * 2) if do_classifier_free_guidance else latents_tile latent_model_input_tile = torch.cat([latents_tile] * 2) if do_classifier_free_guidance else latents_tile
latent_model_input_tile = self.scheduler.scale_model_input(latent_model_input_tile, t) latent_model_input_tile = self.scheduler.scale_model_input(latent_model_input_tile, t)
@ -730,8 +728,6 @@ class CogVideoX_Fun_Pipeline_Control(VideoSysPipeline):
input_start_t = ofs_t input_start_t = ofs_t
input_end_t = ofs_t + t_tile_length input_end_t = ofs_t + t_tile_length
print("input_start_t, input_end_t", input_start_t, input_end_t, latents_all.shape)
print("t_tile_weights.shape", t_tile_weights.shape)
latents_all[:, input_start_t:input_end_t,:, :, :] += latents_all_list[t_i] * t_tile_weights latents_all[:, input_start_t:input_end_t,:, :, :] += latents_all_list[t_i] * t_tile_weights
contributors[:, input_start_t:input_end_t,:, :, :] += t_tile_weights contributors[:, input_start_t:input_end_t,:, :, :] += t_tile_weights