From 20dfe1b4d56bbcfe45b78234d473a90dd7d21ec6 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 18 Sep 2025 08:22:04 -0600 Subject: [PATCH] Small double tap of detach on qwen just for good measure --- .../diffusion_models/qwen_image/qwen_image.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py index 07dcde15..edfe0c13 100644 --- a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py +++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py @@ -284,11 +284,11 @@ class QwenImageModel(BaseModel): txt_seq_lens = prompt_embeds_mask.sum(dim=1).tolist() noise_pred = self.transformer( - hidden_states=latent_model_input.to(self.device_torch, self.torch_dtype), - timestep=timestep / 1000, + hidden_states=latent_model_input.to(self.device_torch, self.torch_dtype).detach(), + timestep=(timestep / 1000).detach(), guidance=None, - encoder_hidden_states=enc_hs, - encoder_hidden_states_mask=prompt_embeds_mask, + encoder_hidden_states=enc_hs.detach(), + encoder_hidden_states_mask=prompt_embeds_mask.detach(), img_shapes=img_shapes, txt_seq_lens=txt_seq_lens, return_dict=False,