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

Add Diffusion4k stride to partitioned_vae

Browse Source
This commit is contained in:
rockerBOO
2025-05-08 19:12:10 -04:00
parent a38e255939
commit baa48d0754
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
flux_train_network.py
View File

@@ -144,6 +144,7 @@ class FluxNetworkTrainer(train_network.NetworkTrainer):
# Apply partitioned for Diffusion4k
if args.partitioned_vae:
ae.decoder.partitioned = True
ae.decoder.stride = 2 # Diffusion4k stride
return flux_utils.MODEL_VERSION_FLUX_V1, [clip_l, t5xxl], ae, model