wasmdashai commited on
Commit
b13d3e8
·
verified ·
1 Parent(s): 635dab7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -521,7 +521,7 @@ class TrinerModelVITS:
521
  torch.nn.utils.remove_weight_norm(flow.conv_pre)
522
  torch.nn.utils.remove_weight_norm(flow.conv_post)
523
 
524
- self.input_save_pretrained(path_save_model,token=self.token)
525
 
526
 
527
  def run_train_epoch(self):
 
521
  torch.nn.utils.remove_weight_norm(flow.conv_pre)
522
  torch.nn.utils.remove_weight_norm(flow.conv_post)
523
 
524
+ model.push_to_hub(path_save_model,token=self.token)
525
 
526
 
527
  def run_train_epoch(self):