Commit 96534194 authored by Dan Lyth's avatar Dan Lyth
Browse files

moving new scripts to 'training'

parent 11c9070d
...@@ -54,10 +54,10 @@ from parler_tts import ( ...@@ -54,10 +54,10 @@ from parler_tts import (
build_delay_pattern_mask, build_delay_pattern_mask,
) )
from parler_tts.utils import get_last_checkpoint, rotate_checkpoints, log_pred, log_metric from training.utils import get_last_checkpoint, rotate_checkpoints, log_pred, log_metric
from parler_tts.arguments import ModelArguments, DataTrainingArguments, ParlerTTSTrainingArguments from training.arguments import ModelArguments, DataTrainingArguments, ParlerTTSTrainingArguments
from parler_tts.data import load_multiple_datasets, DataCollatorParlerTTSWithPadding, DataCollatorEncodecWithPadding from training.data import load_multiple_datasets, DataCollatorParlerTTSWithPadding, DataCollatorEncodecWithPadding
from parler_tts.eval import clap_similarity, wer from training.eval import clap_similarity, wer
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -1014,4 +1014,4 @@ def main(): ...@@ -1014,4 +1014,4 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
set_start_method("spawn") set_start_method("spawn")
main() main()
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment