From 4f825bc07077cefb6c30143f7af24e308a67557a Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Thu, 27 Feb 2025 22:01:35 +0000 Subject: [PATCH] fix zero'd cond_t5 device (#2701) 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 --- backend/diffusion_engine/sd35.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/diffusion_engine/sd35.py b/backend/diffusion_engine/sd35.py index 711fb442..5bf2fc72 100644 --- a/backend/diffusion_engine/sd35.py +++ b/backend/diffusion_engine/sd35.py @@ -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)