diff --git a/TTS/trainer.py b/TTS/trainer.py index 4ca95fba..68b45fe2 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -328,7 +328,7 @@ class Trainer: return obj print(" > Restoring from %s ..." % os.path.basename(restore_path)) - checkpoint = load_fsspec(restore_path, map_location='cpu') + checkpoint = load_fsspec(restore_path, map_location="cpu") try: print(" > Restoring Model...") model.load_state_dict(checkpoint["model"])