r3gm commited on
Commit
cd7a065
1 Parent(s): 452ef74

Update app_rvc.py

Browse files

fix cuda onnxruntime

Files changed (1) hide show
  1. app_rvc.py +4 -1
app_rvc.py CHANGED
@@ -4,6 +4,10 @@ os.system("pip install -q piper-tts==1.2.0")
4
  os.system("pip install -q -r requirements_xtts.txt")
5
  os.system("pip install -q TTS==0.21.1 --no-deps")
6
  import spaces
 
 
 
 
7
  import librosa
8
  from soni_translate.logging_setup import (
9
  logger,
@@ -11,7 +15,6 @@ from soni_translate.logging_setup import (
11
  configure_logging_libs,
12
  ); configure_logging_libs() # noqa
13
  import whisperx
14
- import torch
15
  import os
16
  from soni_translate.audio_segments import create_translated_audio
17
  from soni_translate.text_to_speech import (
 
4
  os.system("pip install -q -r requirements_xtts.txt")
5
  os.system("pip install -q TTS==0.21.1 --no-deps")
6
  import spaces
7
+ import torch
8
+ if os.environ.get("ZERO_GPU") != "TRUE" and torch.cuda.is_available():
9
+ # onnxruntime GPU
10
+ os.system("pip install ort-nightly-gpu --index-url=https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/")
11
  import librosa
12
  from soni_translate.logging_setup import (
13
  logger,
 
15
  configure_logging_libs,
16
  ); configure_logging_libs() # noqa
17
  import whisperx
 
18
  import os
19
  from soni_translate.audio_segments import create_translated_audio
20
  from soni_translate.text_to_speech import (