Spaces:

wasmdashai
/

RunTasking

Running on Zero

wasmdashai commited on Aug 21, 2024

Commit

ad16f6e

verified ·

1 Parent(s): 2835be9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -600,7 +600,9 @@ def init_training():
 @spaces.GPU
 def init_Starting():
-    globals pro
     pro=TrinerModelVITS(dir_model=dir_model,
                     path_training_args='VitsModelSplit/finetune_config_ara.json',
                     train_dataset_dirs = train_dataset_dirs,

 @spaces.GPU
 def init_Starting():
+    global pro
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     pro=TrinerModelVITS(dir_model=dir_model,
                     path_training_args='VitsModelSplit/finetune_config_ara.json',
                     train_dataset_dirs = train_dataset_dirs,