Commit
•
8a2c183
1
Parent(s):
a15bc89
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def GenerateMusic():
|
|
| 36 |
|
| 37 |
SEQ_LEN = 2048
|
| 38 |
PAD_IDX = 780
|
| 39 |
-
DEVICE = '
|
| 40 |
|
| 41 |
# instantiate the model
|
| 42 |
|
|
@@ -110,7 +110,7 @@ def GenerateMusic():
|
|
| 110 |
|
| 111 |
inp = [outy] * number_of_batches_to_generate
|
| 112 |
|
| 113 |
-
inp = torch.LongTensor(inp).
|
| 114 |
|
| 115 |
with ctx:
|
| 116 |
out = model.generate(inp,
|
|
|
|
| 36 |
|
| 37 |
SEQ_LEN = 2048
|
| 38 |
PAD_IDX = 780
|
| 39 |
+
DEVICE = 'cpu' # 'cuda'
|
| 40 |
|
| 41 |
# instantiate the model
|
| 42 |
|
|
|
|
| 110 |
|
| 111 |
inp = [outy] * number_of_batches_to_generate
|
| 112 |
|
| 113 |
+
inp = torch.LongTensor(inp).cpu()
|
| 114 |
|
| 115 |
with ctx:
|
| 116 |
out = model.generate(inp,
|