github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-03 13:05:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

ruff complained

Browse Source
This commit is contained in:
David Burnett
2024-11-08 10:31:53 +00:00
parent bb3cedddd5
commit 2618ed0ae7
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/invocations/flux_vae_encode.py
View File

@@ -45,9 +45,7 @@ class FluxVaeEncodeInvocation(BaseInvocation):
with vae_info as vae:
assert isinstance(vae, AutoEncoder)
vae_dtype = next(iter(vae.state_dict().items()))[1].dtype
image_tensor = image_tensor.to(
device=TorchDevice.choose_torch_device(), dtype=vae_dtype
)
image_tensor = image_tensor.to(device=TorchDevice.choose_torch_device(), dtype=vae_dtype)
latents = vae.encode(image_tensor, sample=True, generator=generator)
return latents