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

Merge pull request #1525 from Akegarasu/sd3

Browse Source 
make guidance_scale keep float in args
This commit is contained in:
Kohya S.
2024-08-29 22:08:57 +09:00
committed by GitHub
parent daa6ad5165 6c0e8a5a17
commit 930d709e3d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
flux_train_network.py
View File

@@ -324,7 +324,8 @@ class FluxNetworkTrainer(train_network.NetworkTrainer):
img_ids = flux_utils.prepare_img_ids(bsz, packed_latent_height, packed_latent_width).to(device=accelerator.device)
# get guidance
guidance_vec = torch.full((bsz,), args.guidance_scale, device=accelerator.device)
# ensure guidance_scale in args is float
guidance_vec = torch.full((bsz,), float(args.guidance_scale), device=accelerator.device)
# ensure the hidden state will require grad
if args.gradient_checkpointing: