Update models.py

models.py

@@ -635,10 +635,5 @@ class AudioLDM2Wrapper(PipelineWrapper):
 def load_model(model_id, device, num_diffusion_steps, double_precision=False):
     ldm_stable = AudioLDM2Wrapper(model_id=model_id, device=device, double_precision=double_precision)
     ldm_stable.load_scheduler()
-    ldm_stable.model.scheduler.set_timesteps(num_diffusion_steps, device=device)
     torch.cuda.empty_cache()
-    # controller = AttentionStore()
-    # controller = EmptyControl()
-    # register_attention_control(ldm_stable.model, controller)
-    # return ldm_stable, controller
     return ldm_stable