Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -585,14 +585,7 @@ train_dataset_dirs=[
|
|
585 |
|
586 |
|
587 |
dir_model='wasmdashai/vits-ar-huba-fine'
|
588 |
-
pro=
|
589 |
-
path_training_args='VitsModelSplit/finetune_config_ara.json',
|
590 |
-
train_dataset_dirs = train_dataset_dirs,
|
591 |
-
eval_dataset_dir = os.path.join(dataset_dir,'eval'),
|
592 |
-
full_generation_dir = os.path.join(dataset_dir,'full_generation'),
|
593 |
-
token=token,
|
594 |
-
device=device
|
595 |
-
)
|
596 |
|
597 |
@spaces.GPU
|
598 |
def run_train_epoch(num):
|
@@ -607,7 +600,15 @@ def init_training():
|
|
607 |
|
608 |
@spaces.GPU
|
609 |
def init_Starting():
|
610 |
-
pro
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
611 |
return 'init_Starting'
|
612 |
@spaces.GPU
|
613 |
def init_wandb():
|
|
|
585 |
|
586 |
|
587 |
dir_model='wasmdashai/vits-ar-huba-fine'
|
588 |
+
pro=None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
589 |
|
590 |
@spaces.GPU
|
591 |
def run_train_epoch(num):
|
|
|
600 |
|
601 |
@spaces.GPU
|
602 |
def init_Starting():
|
603 |
+
globals pro
|
604 |
+
pro=TrinerModelVITS(dir_model=dir_model,
|
605 |
+
path_training_args='VitsModelSplit/finetune_config_ara.json',
|
606 |
+
train_dataset_dirs = train_dataset_dirs,
|
607 |
+
eval_dataset_dir = os.path.join(dataset_dir,'eval'),
|
608 |
+
full_generation_dir = os.path.join(dataset_dir,'full_generation'),
|
609 |
+
token=token,
|
610 |
+
device=device
|
611 |
+
)
|
612 |
return 'init_Starting'
|
613 |
@spaces.GPU
|
614 |
def init_wandb():
|