KingNish commited on
Commit
79a96f2
·
verified ·
1 Parent(s): f85e989

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ from models.soundstream_hubert_new import SoundStream
70
  from vocoder import build_codec_model, process_audio
71
  from post_process_audio import replace_low_freq_with_energy_matched
72
 
73
- device = "cuda:0"
74
 
75
  model = AutoModelForCausalLM.from_pretrained(
76
  "m-a-p/YuE-s1-7B-anneal-en-cot",
 
70
  from vocoder import build_codec_model, process_audio
71
  from post_process_audio import replace_low_freq_with_energy_matched
72
 
73
+ device = "cuda"
74
 
75
  model = AutoModelForCausalLM.from_pretrained(
76
  "m-a-p/YuE-s1-7B-anneal-en-cot",