zero init
Browse files
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 |
|