Spaces:

haoheliu
/

audioldm-text-to-audio-generation

Running on A10G

haoheliu commited on Sep 1

Commit

dcb6d82

•

1 Parent(s): bf5e260

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from transformers import AutoProcessor, ClapModel
 # make Space compatible with CPU duplicates
 if torch.cuda.is_available():
     device = "cuda"
-    torch_dtype = torch.float16
 else:
     device = "cpu"
     torch_dtype = torch.float32

 # make Space compatible with CPU duplicates
 if torch.cuda.is_available():
     device = "cuda"
+    torch_dtype = torch.float32
 else:
     device = "cpu"
     torch_dtype = torch.float32