From d8db618de0d943b1d2b0caeb85de1fe89d412fa4 Mon Sep 17 00:00:00 2001
From: mickr777 <115216705+mickr777@users.noreply.github.com>
Date: Wed, 19 Jul 2023 16:43:02 +1000
Subject: [PATCH] import choose_torch_device from ...backend.util.devices

---
 invokeai/app/invocations/latent.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py
index 28613b1a4d..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,10 +38,6 @@ from diffusers.models.attention_processor import (
     XFormersAttnProcessor,
 )
 
-def choose_torch_device() -> str:
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    return device
-
 class LatentsField(BaseModel):
     """A latents field used for passing latents between invocations"""