Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -127,7 +127,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
|
|
| 127 |
|
| 128 |
inp = torch.LongTensor([outy]).cuda()
|
| 129 |
|
| 130 |
-
with torch.amp.autocast(device_type='cuda', dtype=torch.float16)
|
| 131 |
out = model.module.generate(inp,
|
| 132 |
1,
|
| 133 |
temperature=0.9,
|
|
|
|
| 127 |
|
| 128 |
inp = torch.LongTensor([outy]).cuda()
|
| 129 |
|
| 130 |
+
with torch.amp.autocast(device_type='cuda', dtype=torch.float16):
|
| 131 |
out = model.module.generate(inp,
|
| 132 |
1,
|
| 133 |
temperature=0.9,
|