GrandaddyShmax commited on
Commit
d170ebc
1 Parent(s): b0d9cc0

Update audiocraft/models/loaders.py

Browse files
Files changed (1) hide show
  1. audiocraft/models/loaders.py +4 -4
audiocraft/models/loaders.py CHANGED
@@ -31,10 +31,10 @@ from . import builders
31
 
32
 
33
  HF_MODEL_CHECKPOINTS_MAP = {
34
- "small": "facebook/musicgen-small",
35
- "medium": "facebook/musicgen-medium",
36
- "large": "facebook/musicgen-large",
37
- "melody": "facebook/musicgen-melody",
38
  }
39
 
40
 
 
31
 
32
 
33
  HF_MODEL_CHECKPOINTS_MAP = {
34
+ "small": "GrandaddyShmax/musicgen-small",
35
+ "medium": "GrandaddyShmax/musicgen-medium",
36
+ "large": "GrandaddyShmax/musicgen-large",
37
+ "melody": "GrandaddyShmax/musicgen-melody",
38
  }
39
 
40