wasmdashai commited on
Commit
9d09e8d
·
verified ·
1 Parent(s): 71ab67f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -503,6 +503,8 @@ class TrinerModelVITS:
503
  training_args.weight_mel=45
504
  training_args.num_train_epochs=4
505
  training_args.eval_steps=1000
 
 
506
 
507
  set_seed(training_args.seed)
508
  # scaler = GradScaler(enabled=training_args.fp16)
 
503
  training_args.weight_mel=45
504
  training_args.num_train_epochs=4
505
  training_args.eval_steps=1000
506
+ training_args.fp16=True
507
+
508
 
509
  set_seed(training_args.seed)
510
  # scaler = GradScaler(enabled=training_args.fp16)