George870 Fabrice-TIERCELIN commited on
Commit
6ac331f
1 Parent(s): 05f1747

This Pull Request fixes the space by using a reacheable model (#1)

Browse files

- This Pull Request fixes the space by using a reacheable model (fe669e9d8ce3d26d56d01f3e2e8b3eb83ed1658c)


Co-authored-by: Fabrice TIERCELIN <[email protected]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from inversion_utils import inversion_forward_process, inversion_reverse_process
20
  LDM2 = "cvssp/audioldm2"
21
  MUSIC = "cvssp/audioldm2-music"
22
  LDM2_LARGE = "cvssp/audioldm2-large"
23
- STABLEAUD = "stabilityai/stable-audio-open-1.0"
24
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
  ldm2 = load_model(model_id=LDM2, device=device)
26
  ldm2_large = load_model(model_id=LDM2_LARGE, device=device)
 
20
  LDM2 = "cvssp/audioldm2"
21
  MUSIC = "cvssp/audioldm2-music"
22
  LDM2_LARGE = "cvssp/audioldm2-large"
23
+ STABLEAUD = "chaowenguo/stable-audio-open-1.0"
24
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
  ldm2 = load_model(model_id=LDM2, device=device)
26
  ldm2_large = load_model(model_id=LDM2_LARGE, device=device)