mrfakename
commited on
Commit
•
9a3e0d5
1
Parent(s):
01eed3a
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
- model/trainer.py +2 -0
model/trainer.py
CHANGED
@@ -295,4 +295,6 @@ class Trainer:
|
|
295 |
if global_step % self.last_per_steps == 0:
|
296 |
self.save_checkpoint(global_step, last=True)
|
297 |
|
|
|
|
|
298 |
self.accelerator.end_training()
|
|
|
295 |
if global_step % self.last_per_steps == 0:
|
296 |
self.save_checkpoint(global_step, last=True)
|
297 |
|
298 |
+
self.save_checkpoint(global_step, last=True)
|
299 |
+
|
300 |
self.accelerator.end_training()
|