diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py
index 2800624235..743a648785 100644
--- a/invokeai/app/invocations/latent.py
+++ b/invokeai/app/invocations/latent.py
@@ -22,7 +22,7 @@ from ...backend.stable_diffusion.diffusers_pipeline import (
 from ...backend.stable_diffusion.diffusion.shared_invokeai_diffusion import \
     PostprocessingSettings
 from ...backend.stable_diffusion.schedulers import SCHEDULER_MAP
-from ...backend.util.devices import torch_dtype
+from ...backend.util.devices import choose_torch_device, torch_dtype
 from ..models.image import ImageCategory, ImageField, ResourceOrigin
 from .baseinvocation import (BaseInvocation, BaseInvocationOutput,
                              InvocationConfig, InvocationContext)
@@ -38,7 +38,6 @@ from diffusers.models.attention_processor import (
     XFormersAttnProcessor,
 )
 
-
 class LatentsField(BaseModel):
     """A latents field used for passing latents between invocations"""