Hugging Face's logo

Spaces:
mysome
/
YuE-music-generator-demo
Runtime error

App Files Community
mysome commited on
Commit
587ef66
·
verified ·
1 Parent(s): 720d9d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ from models.soundstream_hubert_new import SoundStream
50
  print("🧠 Loading YuE model on CPU (small version)...")
51
  try:
52
  model = AutoModelForCausalLM.from_pretrained(
53
- "m-a-p/YuE-s1-1B-mini-en-cot",
54
  torch_dtype=torch.float32,
55
  attn_implementation="eager",
56
  low_cpu_mem_usage=True,
 
50
  print("🧠 Loading YuE model on CPU (small version)...")
51
  try:
52
  model = AutoModelForCausalLM.from_pretrained(
53
+ "m-a-p/YuE-upsampler",
54
  torch_dtype=torch.float32,
55
  attn_implementation="eager",
56
  low_cpu_mem_usage=True,