1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix type error in finetune_gui.py

pull/3456/head
suzushi 2025-11-03 13:20:55 +01:00
parent 4161d1d80a
commit e1d53ea5cc
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
kohya_gui/finetune_gui.py
View File

@ -875,12 +875,14 @@ def train_model(
log.info(max_train_steps_info) log.info(max_train_steps_info)
# Calculate lr_warmup_steps # Calculate lr_warmup_steps
lr_warmup = float(lr_warmup) if lr_warmup not in ("", None) else 0
lr_warmup_steps = int(lr_warmup_steps) if lr_warmup_steps not in ("", None) else 0
if lr_warmup_steps > 0: if lr_warmup_steps > 0:
lr_warmup_steps = int(lr_warmup_steps)
if lr_warmup > 0: if lr_warmup > 0:
log.warning("Both lr_warmup and lr_warmup_steps are set. lr_warmup_steps will be used.") log.warning("Both lr_warmup and lr_warmup_steps are set. lr_warmup_steps will be used.")
elif lr_warmup != 0: elif lr_warmup != 0:
lr_warmup_steps = lr_warmup / 100 lr_warmup_steps = int(lr_warmup * max_train_steps / 100) if max_train_steps > 0 else 0
else: else:
lr_warmup_steps = 0 lr_warmup_steps = 0