szin94 commited on
Commit
f33782e
1 Parent(s): 1571184
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -120,8 +120,9 @@ def run(duration, resolution, pitch, glissando, vibrato, stiffness, tension, plu
120
 
121
  params, f_0, u_0 = get_data( \
122
  duration, resolution, pitch, glissando, vibrato, stiffness, tension, pluck, amplitude)
 
123
  if torch.cuda.is_available():
124
- params = [p.cuda() for p in params]
125
  f_0 = f_0.cuda()
126
  u_0 = u_0.cuda()
127
 
 
120
 
121
  params, f_0, u_0 = get_data( \
122
  duration, resolution, pitch, glissando, vibrato, stiffness, tension, pluck, amplitude)
123
+
124
  if torch.cuda.is_available():
125
+ params = [p.cuda() if p is not None else p for p in params]
126
  f_0 = f_0.cuda()
127
  u_0 = u_0.cuda()
128