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

feat: add guidance rescale options for Adaptive Projected Guidance in inference

Browse Source
This commit is contained in:
Kohya S
2025-09-21 13:03:14 +09:00
parent 9621d9d637
commit 040d976597
3 changed files with 27 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/hunyuan_image_utils.py
View File

@@ -401,8 +401,6 @@ class AdaptiveProjectedGuidance:
guidance_rescale: float = 0.0,
use_original_formulation: bool = False,
):
assert guidance_rescale == 0.0, "guidance_rescale > 0.0 not supported."
self.guidance_scale = guidance_scale
self.adaptive_projected_guidance_momentum = adaptive_projected_guidance_momentum
self.adaptive_projected_guidance_rescale = adaptive_projected_guidance_rescale
@@ -425,6 +423,10 @@ class AdaptiveProjectedGuidance:
self.use_original_formulation,
)
if self.guidance_rescale > 0.0:
print(f"Applying guidance rescale with factor {self.guidance_rescale} at step {step}")
pred = rescale_noise_cfg(pred, pred_cond, self.guidance_rescale)
return pred