finetune_gradio not to use fp16 by default for mps device

This commit is contained in:
SWivid
2025-04-03 22:33:21 +08:00
parent fe5c562212
commit 4b4359bc39

View File

@@ -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):