VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 13:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add 'tak' to recognized words and update block swap method to support backward pass

Browse Source
This commit is contained in:
Kohya S
2025-09-12 22:00:41 +09:00
parent aa0af24d01
commit 7a651efd4d
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
_typos.toml
View File

@@ -31,6 +31,7 @@ wn="wn"
hime="hime"
OT="OT"
byt="byt"
tak="tak"
# [files]
# # Extend the default list of files to check

2
hunyuan_image_train_network.py
View File

@@ -394,7 +394,7 @@ class HunyuanImageNetworkTrainer(train_network.NetworkTrainer):
if self.is_swapping_blocks:
# Swap blocks between CPU and GPU to reduce memory usage, in forward and backward passes.
logger.info(f"enable block swap: blocks_to_swap={args.blocks_to_swap}")
model.enable_block_swap(args.blocks_to_swap, accelerator.device)
model.enable_block_swap(args.blocks_to_swap, accelerator.device, supports_backward=True)
return model, text_encoders