wasmdashai commited on
Commit
1a6cc6d
·
verified ·
1 Parent(s): a31b12c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ import os
28
  from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
29
  from VitsModelSplit.PosteriorDecoderModel import PosteriorDecoderModel
30
  from VitsModelSplit.feature_extraction import VitsFeatureExtractor
31
-
32
  from transformers import AutoTokenizer, HfArgumentParser, set_seed
33
  from VitsModelSplit.Arguments import DataTrainingArguments, ModelArguments, VITSTrainingArguments
34
  from VitsModelSplit.dataset_features_collector import FeaturesCollectionDataset
 
28
  from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
29
  from VitsModelSplit.PosteriorDecoderModel import PosteriorDecoderModel
30
  from VitsModelSplit.feature_extraction import VitsFeatureExtractor
31
+ from VitsModelSplit.vits_models_only_decoder imports Vits_models_only_decoder
32
  from transformers import AutoTokenizer, HfArgumentParser, set_seed
33
  from VitsModelSplit.Arguments import DataTrainingArguments, ModelArguments, VITSTrainingArguments
34
  from VitsModelSplit.dataset_features_collector import FeaturesCollectionDataset