Spaces:

wasmdashai
/

RunTasking

Running on Zero

wasmdashai commited on Aug 24, 2024

Commit

366fc7f

verified ·

1 Parent(s): ed1d65a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -545,6 +545,7 @@ class TrinerModelVITS:
         # self.tools=load_tools()
         # self.stute_mode=True
         # print(self.lr_schedulers)
         self.optimizers=(optimizer.cuda(),disc_optimizer.cuda(),scaler)

         # self.tools=load_tools()
         # self.stute_mode=True
         # print(self.lr_schedulers)
+        optimizer,disc_optimizer,scaler=self.optimizers
         self.optimizers=(optimizer.cuda(),disc_optimizer.cuda(),scaler)