Spaces:

wasmdashai
/

RunTasking

Running on Zero

wasmdashai commited on Aug 21, 2024

Commit

f33e37e

verified ·

1 Parent(s): 36ef269

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -450,7 +450,7 @@ class TrinerModelVITS:
     def initialize_training_components(self):
         training_args= self.training_args
         training_args.weight_kl=1
         training_args.d_learning_rate=2e-4

     def initialize_training_components(self):
+        self.training_args=load_training_args(self.path_training_args)
         training_args= self.training_args
         training_args.weight_kl=1
         training_args.d_learning_rate=2e-4