ylacombe commited on
Commit
ae4900f
·
verified ·
1 Parent(s): 77942a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from transformers.generation.streamers import BaseStreamer
17
  device = "cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
18
  torch_dtype = torch.float16 if device != "cpu" else torch.float32
19
 
20
- repo_id = "ylacombe/parler-tts-mini-v1"
21
 
22
  model = ParlerTTSForConditionalGeneration.from_pretrained(
23
  repo_id, torch_dtype=torch_dtype, attn_implementation="eager",
 
17
  device = "cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
18
  torch_dtype = torch.float16 if device != "cpu" else torch.float32
19
 
20
+ repo_id = "parler-tts/parler_tts_mini_v0.1"
21
 
22
  model = ParlerTTSForConditionalGeneration.from_pretrained(
23
  repo_id, torch_dtype=torch_dtype, attn_implementation="eager",