diff --git a/TTS/utils/arguments.py b/TTS/utils/arguments.py
index 6a09986c..dc0a6764 100644
--- a/TTS/utils/arguments.py
+++ b/TTS/utils/arguments.py
@@ -6,9 +6,11 @@ import argparse
 import glob
 import os
 import re
+import torch
 
 from TTS.tts.utils.generic_utils import check_config_tts
 from TTS.tts.utils.text.symbols import parse_symbols
+from TTS.tts.utils.io import load_checkpoint
 from TTS.utils.console_logger import ConsoleLogger
 from TTS.utils.generic_utils import create_experiment_folder, get_git_branch
 from TTS.utils.io import copy_model_files, load_config