gdnartea commited on
Commit
2e41b4d
·
verified ·
1 Parent(s): f65ce4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -19,7 +19,7 @@ proc_model_name = "microsoft/Phi-3-mini-4k-instruct"
19
  proc_model = AutoModelForCausalLM.from_pretrained(
20
  proc_model_name,
21
  torch_dtype=torch.float16,
22
- revision=653ee820c4f2ee66427e997b4a8ca3e9323e7d46,
23
  trust_remote_code=True,
24
  low_cpu_mem_usage=True,
25
  )
@@ -41,9 +41,10 @@ decoding_cfg.beam.beam_size = 1
41
  model.change_decoding_strategy(decoding_cfg)
42
 
43
 
44
-
45
- vits_model = VitsModel.from_pretrained("facebook/mms-tts-eng")
46
- vits_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
 
47
  set_seed(555)
48
 
49
 
 
19
  proc_model = AutoModelForCausalLM.from_pretrained(
20
  proc_model_name,
21
  torch_dtype=torch.float16,
22
+ revision="653ee820c4f2ee66427e997b4a8ca3e9323e7d46",
23
  trust_remote_code=True,
24
  low_cpu_mem_usage=True,
25
  )
 
41
  model.change_decoding_strategy(decoding_cfg)
42
 
43
 
44
+ #vits_model_name = "facebook/mms-tts-eng"
45
+ vits_model_name = "coqui/XTTS-v2"
46
+ vits_model = VitsModel.from_pretrained(vits_model_name)
47
+ vits_tokenizer = AutoTokenizer.from_pretrained()
48
  set_seed(555)
49
 
50