lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-10 23:49:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

distilled cfg scale

Browse Source
This commit is contained in:
layerdiffusion
2024-08-08 00:57:40 -07:00
parent 653be0a3ad
commit 572a3c3d8b
6 changed files with 25 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/diffusion_engine/flux.py
View File

@@ -78,10 +78,13 @@ class Flux(ForgeDiffusionEngine):
cond_l, pooled_l = self.text_processing_engine_l(prompt)
cond_t5 = self.text_processing_engine_t5(prompt)
distilled_cfg_scale = getattr(prompt, 'distilled_cfg_scale', 3.5) or 3.5
print(f'distilled_cfg_scale = {distilled_cfg_scale}')
cond = dict(
crossattn=cond_t5,
vector=pooled_l,
guidance=torch.FloatTensor([3.5] * len(prompt))
guidance=torch.FloatTensor([distilled_cfg_scale] * len(prompt))
)
return cond