From 92d2d337fd98673c126fd40727e067204e4523ae Mon Sep 17 00:00:00 2001
From: Spr_Aachen <51275522+Spr-Aachen@users.noreply.github.com>
Date: Fri, 28 Nov 2025 22:53:43 +0800
Subject: [PATCH] Fix training error caused by float type of default_batch_size
 parameter (#2662)

---
 webui.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/webui.py b/webui.py
index beb0963a..9c5ddd07 100644
--- a/webui.py
+++ b/webui.py
@@ -116,8 +116,8 @@ def set_default():
     gpu_info = "\n".join(gpu_infos)
     if is_gpu_ok:
         minmem = min(mem)
-        default_batch_size = minmem // 2 if version not in v3v4set else minmem // 8
-        default_batch_size_s1 = minmem // 2
+        default_batch_size = int(minmem // 2 if version not in v3v4set else minmem // 8)
+        default_batch_size_s1 = int(minmem // 2)
     else:
         default_batch_size = default_batch_size_s1 = int(psutil.virtual_memory().total / 1024 / 1024 / 1024 / 4)
     if version not in v3v4set: