mrfakename commited on
Commit
3ce666d
1 Parent(s): 74dee3d

Sync from GitHub repo

Browse files

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

Files changed (1) hide show
  1. src/f5_tts/model/trainer.py +1 -1
src/f5_tts/model/trainer.py CHANGED
@@ -148,7 +148,7 @@ class Trainer:
148
  if (
149
  not exists(self.checkpoint_path)
150
  or not os.path.exists(self.checkpoint_path)
151
- or not os.listdir(self.checkpoint_path)
152
  ):
153
  return 0
154
 
 
148
  if (
149
  not exists(self.checkpoint_path)
150
  or not os.path.exists(self.checkpoint_path)
151
+ or not any(filename.endswith(".pt") for filename in os.listdir(self.checkpoint_path))
152
  ):
153
  return 0
154