gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-05 20:40:12 -08:00
Code Issues Packages Projects Releases Wiki Activity

last fix patch-1

Browse Source
This commit is contained in:
SWivid
2025-07-08 18:44:47 +08:00
parent efc7a7498b
commit a275798a2f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/f5_tts/model/trainer.py
View File

@@ -242,7 +242,7 @@ class Trainer:
del checkpoint["model_state_dict"][key] del checkpoint["model_state_dict"][key]
self.accelerator.unwrap_model(self.model).load_state_dict(checkpoint["model_state_dict"]) self.accelerator.unwrap_model(self.model).load_state_dict(checkpoint["model_state_dict"])
self.accelerator.unwrap_model(self.optimizer).load_state_dict(checkpoint["optimizer_state_dict"]) self.optimizer.load_state_dict(checkpoint["optimizer_state_dict"])
if self.scheduler: if self.scheduler:
self.scheduler.load_state_dict(checkpoint["scheduler_state_dict"]) self.scheduler.load_state_dict(checkpoint["scheduler_state_dict"])
update = checkpoint["update"] update = checkpoint["update"]