Hugging Face's logo

Spaces:
asigalov61
/
Descriptive-Music-Transformer
Sleeping

App Files Community
1
asigalov61 commited on
Commit
f6be918
·
verified ·
1 Parent(s): 8985974

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def GenerateMusic(input_title):
60
  if DEVICE == 'cpu':
61
  dtype = torch.bfloat16
62
  else:
63
- dtype = torch.float16
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