Update app.py
Browse files
app.py
CHANGED
|
@@ -47,7 +47,7 @@ def Generate_POP_Medley(input_num_medley_comps, input_melody_patch):
|
|
| 47 |
DIM = 64
|
| 48 |
CHANS = 1
|
| 49 |
TSTEPS = 1000
|
| 50 |
-
DEVICE = '
|
| 51 |
|
| 52 |
unet = Unet(
|
| 53 |
dim = DIM,
|
|
|
|
| 47 |
DIM = 64
|
| 48 |
CHANS = 1
|
| 49 |
TSTEPS = 1000
|
| 50 |
+
DEVICE = 'cpu' # 'cuda'
|
| 51 |
|
| 52 |
unet = Unet(
|
| 53 |
dim = DIM,
|