diff --git a/TTS/demos/xtts_ft_demo/utils/gpt_train.py b/TTS/demos/xtts_ft_demo/utils/gpt_train.py index 4d33d6fc..a80370bc 100644 --- a/TTS/demos/xtts_ft_demo/utils/gpt_train.py +++ b/TTS/demos/xtts_ft_demo/utils/gpt_train.py @@ -165,6 +165,6 @@ def train_gpt(language, num_epochs, batch_size, train_csv, eval_csv, output_path speaker_ref = train_samples[longest_text_idx]["audio_file"] # deallocate VRAM - del model, trainer + del model return XTTS_CONFIG_FILE, XTTS_CHECKPOINT, TOKENIZER_FILE, trainer.output_path, speaker_ref