From 4b4359bc39ec9a3dfb6ff412345030792f5c37fb Mon Sep 17 00:00:00 2001 From: SWivid Date: Thu, 3 Apr 2025 22:33:21 +0800 Subject: [PATCH] finetune_gradio not to use fp16 by default for mps device --- src/f5_tts/train/finetune_gradio.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/f5_tts/train/finetune_gradio.py b/src/f5_tts/train/finetune_gradio.py index 4626a5c..81910da 100644 --- a/src/f5_tts/train/finetune_gradio.py +++ b/src/f5_tts/train/finetune_gradio.py @@ -138,6 +138,8 @@ def load_settings(project_name): "logger": "none", "bnb_optimizer": False, } + if device == "mps": + default_settings["mixed_precision"] = "none" # Load settings from file if it exists if os.path.isfile(file_setting):