alexnasa commited on
Commit
4509e9d
·
verified ·
1 Parent(s): 9ee6fc6

debug code removed

Browse files
Files changed (1) hide show
  1. OmniAvatar/wan_video.py +1 -1
OmniAvatar/wan_video.py CHANGED
@@ -256,7 +256,7 @@ class WanVideoPipeline(BasePipeline):
256
 
257
  # Inference
258
  noise_pred_posi = self.dit(latents, timestep=timestep, **prompt_emb_posi, **image_emb, **audio_emb, **tea_cache_posi, **extra_input)
259
- print(f'noise_pred_posi:{noise_pred_posi.dtype}')
260
  if cfg_scale != 1.0:
261
  audio_emb_uc = {}
262
  for key in audio_emb.keys():
 
256
 
257
  # Inference
258
  noise_pred_posi = self.dit(latents, timestep=timestep, **prompt_emb_posi, **image_emb, **audio_emb, **tea_cache_posi, **extra_input)
259
+
260
  if cfg_scale != 1.0:
261
  audio_emb_uc = {}
262
  for key in audio_emb.keys():