Update audiosr/pipeline.py

audiosr/pipeline.py

@@ -111,7 +111,7 @@ def make_batch_for_super_resolution(input_file, waveform=None, fbank=None):
 def round_up_duration(duration):
     return int(round(duration / 2.5) + 1) * 2.5
 
-@spaces.GPU
+
 def build_model(ckpt_path=None, config=None, device=None, model_name="basic"):
     if device is None or device == "auto":
         if torch.cuda.is_available():
@@ -150,6 +150,7 @@ def build_model(ckpt_path=None, config=None, device=None, model_name="basic"):
 
     return latent_diffusion
 
+@spaces.GPU
 
 def super_resolution(
     latent_diffusion,
@@ -165,12 +166,13 @@ def super_resolution(
 
     batch, duration = make_batch_for_super_resolution(input_file, waveform=waveform)
 
+    with torch.no_grad():
     with torch.no_grad():
         waveform = latent_diffusion.generate_batch(
             batch,
             unconditional_guidance_scale=guidance_scale,
             ddim_steps=ddim_steps,
-            duration=duration,
         )
 
     return waveform
+