Hugging Face's logo

Spaces:
Hugorowan
/
BardJukebox
Paused

App Files Community
Hugorowan commited on
Commit
593ef38
·
1 Parent(s): 5bb9b4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from IPython.display import Audio
8
  # Import the model
9
  from jukebox.models import MODELS
10
 
11
- model = jukebox.make_vqvae(MODELS.5B_LYRICS)
12
 
13
 
14
  # Generate music
 
8
  # Import the model
9
  from jukebox.models import MODELS
10
 
11
+ model = jukebox.make_vqvae(MODELS['5B_LYRICS'], device="cpu")
12
 
13
 
14
  # Generate music