wasmdashai commited on
Commit
7fd4bd3
·
verified ·
1 Parent(s): 4a32672

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer
6
  import torch
7
  import os
8
  from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
9
- import VitsModelSplit.monotonic_align
10
 
11
  token=os.environ.get("key_")
12
  tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
@@ -622,7 +622,7 @@ def greet(text,id):
622
  # dict_state_grad_loss=sgl,
623
  nk=50,
624
  path_save_model=dir_model,
625
- maf=model.monotonic_align_max_path,
626
 
627
  n_back_save_model=3000,
628
  start_speeker=0,
 
6
  import torch
7
  import os
8
  from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
9
+ import VitsModelSplit.monotonic_align as monotonic_align
10
 
11
  token=os.environ.get("key_")
12
  tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
 
622
  # dict_state_grad_loss=sgl,
623
  nk=50,
624
  path_save_model=dir_model,
625
+ maf=monotonic_align.maximum_path,
626
 
627
  n_back_save_model=3000,
628
  start_speeker=0,