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 (
         | 
