Fix check for making sure vae is on the right device.

2025-10-21 14:49:20 -06:00
parent 5123090f6c
commit ff14cd6343
5 changed files with 8 additions and 6 deletions
--- a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py
+++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py
@@ -414,7 +414,7 @@ class QwenImageModel(BaseModel):
            dtype = self.vae_torch_dtype

        # Move to vae to device if on cpu
-        if self.vae.device == "cpu":
+        if self.vae.device == torch.device("cpu"):
            self.vae.to(device)
        self.vae.eval()
        self.vae.requires_grad_(False)