Update app_rvc.py
Browse filesfix cuda onnxruntime
- 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 (
|