wasmdashai commited on
Commit
e8da50c
·
verified ·
1 Parent(s): 6031810

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -575,7 +575,7 @@ ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_datas
575
 
576
 
577
 
578
- raining_args.weight_kl=1
579
  training_args.d_learning_rate=2e-4
580
  training_args.learning_rate=2e-4
581
  training_args.weight_mel=45
@@ -592,7 +592,7 @@ wandb.init(project= 'AZ',config = training_args.to_dict())
592
  for i in range(10000):
593
  # model.train(True)
594
  print(f'clcye epochs ={i}')
595
- model=VitsModel.from_pretrained(dir_model).to(device)
596
  # model.setMfA(monotonic_align.maximum_path)
597
  #dir_model_save=dir_model+'/vend'
598
 
 
575
 
576
 
577
 
578
+ training_args.weight_kl=1
579
  training_args.d_learning_rate=2e-4
580
  training_args.learning_rate=2e-4
581
  training_args.weight_mel=45
 
592
  for i in range(10000):
593
  # model.train(True)
594
  print(f'clcye epochs ={i}')
595
+ model=VitsModel.from_pretrained(dir_model,token=token).to(device)
596
  # model.setMfA(monotonic_align.maximum_path)
597
  #dir_model_save=dir_model+'/vend'
598