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

fix zero'd cond_t5 device (#2701)

Browse Source 
could be created on the wrong device if user has enough VRAM for Forge to run text encoders on GPU.
just move it to the cond_l device
This commit is contained in:
DenOfEquity
2025-02-27 22:01:35 +00:00
committed by GitHub
parent 3ce373939e
commit 4f825bc070
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/diffusion_engine/sd35.py
View File

@@ -107,7 +107,7 @@ class StableDiffusion3(ForgeDiffusionEngine):
if opts.sd3_enable_t5:
cond_t5 = self.text_processing_engine_t5(prompt)
else:
cond_t5 = torch.zeros([len(prompt), 256, 4096])
cond_t5 = torch.zeros([len(prompt), 256, 4096]).to(cond_l.device)
is_negative_prompt = getattr(prompt, 'is_negative_prompt', False)