Spaces:
Running
on
Zero
Running
on
Zero
mrfakename
commited on
Commit
•
3ce666d
1
Parent(s):
74dee3d
Sync from GitHub repo
Browse filesThis Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there
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 |
|