cuda

app.py

@@ -113,9 +113,15 @@ def run(duration, resolution, pitch, glissando, vibrato, stiffness, tension, plu
     checkpoint = filter_state_dict(checkpoint['state_dict'])
     model = Synthesizer(**configs)
     model.load_state_dict(checkpoint)
+    if torch.cuda.is_available():
+        model = model.cuda()
     
     params, f_0, u_0 = get_data( \
         duration, resolution, pitch, glissando, vibrato, stiffness, tension, pluck, amplitude)
+    if torch.cuda.is_available():
+        params = [p.cuda() for p in params]
+        f_0 = f_0.cuda()
+        u_0 = u_0.cuda()
 
     with torch.no_grad():
         ut, mode_input, mode_output = model(params, f_0, u_0)