Spaces:

asigalov61
/

Descriptive-Music-Transformer

Running

asigalov61 commited on Jun 30

Commit

f6be918

•

1 Parent(s): 8985974

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ def GenerateMusic(input_title):
     if DEVICE == 'cpu':
         dtype = torch.bfloat16
     else:
-        dtype = torch.float16
     ctx = torch.amp.autocast(device_type=DEVICE, dtype=dtype)

     if DEVICE == 'cpu':
         dtype = torch.bfloat16
     else:
+        dtype = torch.bfloat16
     ctx = torch.amp.autocast(device_type=DEVICE, dtype=dtype)