Hugging Face's logo

Spaces:
kevinwang676
/
gpt_sovits_demo
Running

App Files Community
litagin commited on
Commit
ef0eeb2
·
1 Parent(s): 26be912

Fix device=cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ from my_utils import load_audio
35
  from text import cleaned_text_to_sequence
36
  from text.cleaner import clean_text
37
 
38
- device = "cuda"
39
  tokenizer = AutoTokenizer.from_pretrained(bert_path)
40
  bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
41
  if is_half == True:
 
35
  from text import cleaned_text_to_sequence
36
  from text.cleaner import clean_text
37
 
38
+ device = "cuda" if torch.cuda.is_available() else "cpu"
39
  tokenizer = AutoTokenizer.from_pretrained(bert_path)
40
  bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
41
  if is_half == True: