Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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",
|