Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,7 +59,7 @@ def GenerateDrums(input_midi, input_num_tokens):
|
|
| 59 |
if DEVICE == 'cpu':
|
| 60 |
dtype = torch.bfloat16
|
| 61 |
else:
|
| 62 |
-
dtype = torch.
|
| 63 |
|
| 64 |
ctx = torch.amp.autocast(device_type=DEVICE, dtype=dtype)
|
| 65 |
|
|
|
|
| 59 |
if DEVICE == 'cpu':
|
| 60 |
dtype = torch.bfloat16
|
| 61 |
else:
|
| 62 |
+
dtype = torch.bfloat16
|
| 63 |
|
| 64 |
ctx = torch.amp.autocast(device_type=DEVICE, dtype=dtype)
|
| 65 |
|