wasmdashai commited on
Commit
be47285
·
verified ·
1 Parent(s): 4687b44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -407,6 +407,7 @@ class TrinerModelVITS:
407
  self.len_dataset=len(self.DataSets['train'])
408
  self.load_model()
409
  self.init_wandb()
 
410
 
411
 
412
 
 
407
  self.len_dataset=len(self.DataSets['train'])
408
  self.load_model()
409
  self.init_wandb()
410
+ scaler = GradScaler(enabled=True)
411
 
412
 
413