Mirrors/VoiceCraft
1
0
Fork 0
mirror of https://github.com/jasonppy/VoiceCraft.git synced 2026-04-03 09:46:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2 from Sewlell/master

Browse Source 
merge colab notebook
This commit is contained in:
zuev-stepan
2024-04-11 15:17:27 +05:00
committed by GitHub
parent 2bd19f2314 623550f808
commit 9344110b0c
3 changed files with 138 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gradio_app.py
View File

@@ -1,6 +1,4 @@
import os
# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# os.environ["CUDA_VISIBLE_DEVICES"] = "0" # for local use
import gradio as gr
import torch
import torchaudio
@@ -14,6 +12,10 @@ import numpy as np
import random
import uuid
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
os.chdir("/content/VoiceCraft-gradio-colab")
os.environ['USER'] = 'aaa'
TMP_PATH = os.getenv("TMP_PATH", "./demo/temp")
device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -596,4 +598,4 @@ with gr.Blocks() as app:
if __name__ == "__main__":
app.launch()
app.launch(share=True)