Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|