changed torchdtype

app.py

@@ -9,9 +9,9 @@ if torch.cuda.is_available():
 else:
     device = "cpu"
     torch_dtype = torch.float32
-
+print(device)
 repo_id = "cvssp/audioldm-m-full"
-pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
+pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
 pipe.unet = torch.compile(pipe.unet)
 
 prompt = "Techno music with a strong, upbeat tempo and high melodic riffs"