Spaces:
Runtime error
Runtime error
osanseviero
commited on
Commit
·
cab646b
1
Parent(s):
f086dd7
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import IPython
|
|
2 |
|
3 |
import sys
|
4 |
import subprocess
|
|
|
5 |
|
6 |
subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "--force-reinstall", "git+https://github.com/osanseviero/tortoise-tts.git"])
|
7 |
|
@@ -16,6 +17,7 @@ import numpy as np
|
|
16 |
import gradio as gr
|
17 |
|
18 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
19 |
|
20 |
# This will download all the models used by Tortoise from the HF hub
|
21 |
tts = TextToSpeech(device="cuda")
|
|
|
2 |
|
3 |
import sys
|
4 |
import subprocess
|
5 |
+
import os
|
6 |
|
7 |
subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "--force-reinstall", "git+https://github.com/osanseviero/tortoise-tts.git"])
|
8 |
|
|
|
17 |
import gradio as gr
|
18 |
|
19 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
20 |
+
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
|
21 |
|
22 |
# This will download all the models used by Tortoise from the HF hub
|
23 |
tts = TextToSpeech(device="cuda")
|