osanseviero HF staff commited on
Commit
cab646b
1 Parent(s): f086dd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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")