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

padding

Browse Source
This commit is contained in:
Haoming
2025-12-24 17:41:50 +08:00
parent a44f84830c
commit 4e6abcaac7
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/diffusion_engine/sdxl.py
+5
View File
@@ -91,6 +91,11 @@ class StableDiffusionXL(ForgeDiffusionEngine):
cond_l = torch.zeros_like(cond_l)
cond_g = torch.zeros_like(cond_g)
# Ensure cond_l and cond_g have the same size
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)
cond = dict(
crossattn=torch.cat([cond_l, cond_g], dim=2),
vector=torch.cat([clip_pooled, flat], dim=1),