lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-09 09:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

ling

Browse Source
This commit is contained in:
layerdiffusion
2024-08-11 17:42:46 -07:00
parent 336b587fc0
commit e72d912118
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/diffusion_engine/flux.py
View File

@@ -77,14 +77,12 @@ class Flux(ForgeDiffusionEngine):
memory_management.load_model_gpu(self.forge_objects.clip.patcher)
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)
if self.use_distilled_cfg_scale:
distilled_cfg_scale = getattr(prompt, 'distilled_cfg_scale', 3.5) or 3.5
cond['guidance'] = torch.FloatTensor([distilled_cfg_scale] * len(prompt))
print(f'Distilled CFG Scale: {distilled_cfg_scale}')
else:
print('Distilled CFG Scale will be ignored for Schnell')