Update pico_model.py

pico_model.py

@@ -179,7 +179,8 @@ class ClapText_Onset_2_Audio_Diffusion(nn.Module):
         num_warmup_steps = len(timesteps) - num_steps * inference_scheduler.order
         progress_bar = tqdm(range(num_steps), disable=disable_progress)
         
-        for i, t in tqdm(enumerate(timesteps)):
+        for i, t in enumerate(timesteps):
+            print(i)
             # expand the latents if we are doing classifier free guidance
             latent_model_input = torch.cat([onset_latents] * 2) if classifier_free_guidance else onset_latents
             latent_model_input = inference_scheduler.scale_model_input(latent_model_input, t)