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 device issue

Browse Source
This commit is contained in:
Jaret Burkett
2025-06-26 19:14:25 -06:00
parent 4f91cb7148
commit 01a3c8a9b1
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py
View File

@@ -403,6 +403,5 @@ class FluxKontextModel(BaseModel):
control_latent = self.encode_images(control_tensor).to(latents.device, latents.dtype) control_latent = self.encode_images(control_tensor).to(latents.device, latents.dtype)
latents = torch.cat((latents, control_latent), dim=1) latents = torch.cat((latents, control_latent), dim=1)
self.vae.to('cpu')
return latents.detach() return latents.detach()