Spaces:

wasmdashai
/

RunTasking

Running on Zero

wasmdashai commited on Aug 24, 2024

Commit

13d7d81

verified ·

1 Parent(s): 366fc7f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -545,8 +545,7 @@ class TrinerModelVITS:
         # 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)

         # self.tools=load_tools()
         # self.stute_mode=True
         # print(self.lr_schedulers)