OmniAICreator commited on
Commit
d879372
·
verified ·
1 Parent(s): 27b1f96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ with safe_open(ckpt_path, framework="pt", device="cpu") as f:
31
  codec_model = XCodec2Model.from_pretrained(
32
  None, config=codec_config, state_dict=ckpt
33
  )
34
- codec_model.eval()
35
 
36
  whisper_turbo_pipe = pipeline(
37
  "automatic-speech-recognition",
 
31
  codec_model = XCodec2Model.from_pretrained(
32
  None, config=codec_config, state_dict=ckpt
33
  )
34
+ codec_model.eval().cuda()
35
 
36
  whisper_turbo_pipe = pipeline(
37
  "automatic-speech-recognition",