From 2618ed0ae7a4e1edb1bf395442a7d7fa06fed78f Mon Sep 17 00:00:00 2001 From: David Burnett Date: Fri, 8 Nov 2024 10:31:53 +0000 Subject: [PATCH] ruff complained --- invokeai/app/invocations/flux_vae_encode.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/invokeai/app/invocations/flux_vae_encode.py b/invokeai/app/invocations/flux_vae_encode.py index 9261c1ee0a..f389985dea 100644 --- a/invokeai/app/invocations/flux_vae_encode.py +++ b/invokeai/app/invocations/flux_vae_encode.py @@ -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