Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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,
|