CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

sage

Browse Source
This commit is contained in:
Haoming 2026-01-27 22:00:49 +08:00
parent 0c3974ce2e
commit 7fddeeba8b
1 changed files with 1 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/diffusion_engine/sdxl.py
View File

@ -73,7 +73,6 @@ class StableDiffusionXL(ForgeDiffusionEngine):
width = getattr(prompt, "width", 1024) or 1024
height = getattr(prompt, "height", 1024) or 1024
is_negative_prompt = getattr(prompt, "is_negative_prompt", False)
crop_w = opts.sdxl_crop_left
crop_h = opts.sdxl_crop_top
@ -84,14 +83,7 @@ class StableDiffusionXL(ForgeDiffusionEngine):
flat = torch.flatten(torch.cat(out)).unsqueeze(dim=0).repeat(clip_pooled.shape[0], 1).to(clip_pooled)
force_zero_negative_prompt = is_negative_prompt and all(x == "" for x in prompt)
if force_zero_negative_prompt:
clip_pooled = torch.zeros_like(clip_pooled)
cond_l = torch.zeros_like(cond_l)
cond_g = torch.zeros_like(cond_g)
# Ensure cond_l and cond_g have the same size
# ensure cond_l and cond_g have the same length
max_len = max(cond_l.shape[1], cond_g.shape[1])
cond_l = torch.cat([cond_l, cond_l.new_zeros(cond_l.size(0), max_len - cond_l.shape[1], cond_l.size(2))], dim=1)
cond_g = torch.cat([cond_g, cond_g.new_zeros(cond_g.size(0), max_len - cond_g.shape[1], cond_g.size(2))], dim=1)
@ -189,12 +181,6 @@ class StableDiffusionXLRefiner(ForgeDiffusionEngine):
flat = torch.flatten(torch.cat(out)).unsqueeze(dim=0).repeat(clip_pooled.shape[0], 1).to(clip_pooled)
force_zero_negative_prompt = is_negative_prompt and all(x == "" for x in prompt)
if force_zero_negative_prompt:
clip_pooled = torch.zeros_like(clip_pooled)
cond_g = torch.zeros_like(cond_g)
cond = dict(
crossattn=cond_g,
vector=torch.cat([clip_pooled, flat], dim=1),