tomxxie commited on
Commit
d845e75
·
1 Parent(s): a2e1ce0

适配zeroGPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -119,7 +119,7 @@ gpu_id = 4
119
  # res_text = model.generate(wavs=feat, wavs_len=feat_lens, prompt=input_prompt)[0]
120
  # print("耿雪龙哈哈:", res_text)
121
  # return res_text, now_file_tmp_path_resample
122
-
123
  def do_decode(input_wav_path, input_prompt):
124
  print(f'input_wav_path= {input_wav_path}, input_prompt= {input_prompt}')
125
  # 省略处理逻辑
 
119
  # res_text = model.generate(wavs=feat, wavs_len=feat_lens, prompt=input_prompt)[0]
120
  # print("耿雪龙哈哈:", res_text)
121
  # return res_text, now_file_tmp_path_resample
122
+ @spaces.GPU
123
  def do_decode(input_wav_path, input_prompt):
124
  print(f'input_wav_path= {input_wav_path}, input_prompt= {input_prompt}')
125
  # 省略处理逻辑