Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
0d875dc
1
Parent(s):
a139ebd
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
|
|
| 30 |
|
| 31 |
SEQ_LEN = 8192 # Models seq len
|
| 32 |
PAD_IDX = 767 # Models pad index
|
| 33 |
-
DEVICE = '
|
| 34 |
|
| 35 |
# instantiate the model
|
| 36 |
|
|
|
|
| 30 |
|
| 31 |
SEQ_LEN = 8192 # Models seq len
|
| 32 |
PAD_IDX = 767 # Models pad index
|
| 33 |
+
DEVICE = 'cuda' # 'cpu'
|
| 34 |
|
| 35 |
# instantiate the model
|
| 36 |
|