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,
|