From b9156c0ad52b89ad3935b1b53e4efcfcb076d43d Mon Sep 17 00:00:00 2001
From: SWivid <swivid@qq.com>
Date: Tue, 25 Mar 2025 07:49:19 +0800
Subject: [PATCH] v1.0.8 fix a fatal bug with log_samples since
 37eb3b50da8fdcb24f8b420b5f91246d85d4f5b5

---
 pyproject.toml              | 2 +-
 src/f5_tts/model/trainer.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 7893c7f..338aec7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "f5-tts"
-version = "1.0.7"
+version = "1.0.8"
 description = "F5-TTS: A Fairytaler that Fakes Fluent and Faithful Speech with Flow Matching"
 readme = "README.md"
 license = {text = "MIT License"}
diff --git a/src/f5_tts/model/trainer.py b/src/f5_tts/model/trainer.py
index 6e3888f..b1bab71 100644
--- a/src/f5_tts/model/trainer.py
+++ b/src/f5_tts/model/trainer.py
@@ -428,6 +428,7 @@ class Trainer:
                         torchaudio.save(
                             f"{log_samples_path}/update_{global_update}_ref.wav", ref_audio, target_sample_rate
                         )
+                        self.model.train()
 
                 if global_update % self.last_per_updates == 0 and self.accelerator.sync_gradients:
                     self.save_checkpoint(global_update, last=True)