ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 19:21:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jaret Burkett
2025-10-21 14:49:20 -06:00
parent 5123090f6c
commit ff14cd6343
5 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/diffusion_models/qwen_image/qwen_image.py
View File

@@ -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)