github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-01 07:04:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

Support cfg_scale list in CogView4Denoise.

Browse Source
This commit is contained in:
Ryan Dick
2025-03-10 19:16:24 +00:00
committed by psychedelicious
parent 30f19dc37a
commit 0939030324
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/invocations/cogview4_denoise.py
View File

@@ -191,8 +191,7 @@ class CogView4DenoiseInvocation(BaseInvocation, WithMetadata, WithBoard):
total_steps = len(timesteps) - 1
# Prepare the CFG scale list.
# TODO(ryand): Implement this.
# cfg_scale = self._prepare_cfg_scale(total_steps)
cfg_scale = self._prepare_cfg_scale(total_steps)
# Load the input latents, if provided.
init_latents = context.tensors.load(self.latents.latents_name) if self.latents else None
@@ -276,7 +275,7 @@ class CogView4DenoiseInvocation(BaseInvocation, WithMetadata, WithBoard):
return_dict=False,
)[0]
noise_pred = noise_pred_uncond + self.cfg_scale * (noise_pred_cond - noise_pred_uncond)
noise_pred = noise_pred_uncond + cfg_scale[step_idx] * (noise_pred_cond - noise_pred_uncond)
else:
noise_pred = noise_pred_cond