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