Commit
•
f6be918
1
Parent(s):
8985974
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ def GenerateMusic(input_title):
|
|
60 |
if DEVICE == 'cpu':
|
61 |
dtype = torch.bfloat16
|
62 |
else:
|
63 |
-
dtype = torch.
|
64 |
|
65 |
ctx = torch.amp.autocast(device_type=DEVICE, dtype=dtype)
|
66 |
|
|
|
60 |
if DEVICE == 'cpu':
|
61 |
dtype = torch.bfloat16
|
62 |
else:
|
63 |
+
dtype = torch.bfloat16
|
64 |
|
65 |
ctx = torch.amp.autocast(device_type=DEVICE, dtype=dtype)
|
66 |
|