make guidance_scale keep float in args

kohya-ss · Aug 29, 2024 · 6c0e8a5 · 6c0e8a5
1 parent a61cf73
commit 6c0e8a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flux_train_network.py b/flux_train_network.py
@@ -324,7 +324,8 @@ def get_noise_pred_and_target(
         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: