From 52e7e0b31bb688789c0715604dd8e98f661e8812 Mon Sep 17 00:00:00 2001 From: mickr777 <115216705+mickr777@users.noreply.github.com> Date: Wed, 19 Jul 2023 15:15:55 +1000 Subject: [PATCH 1/3] Missing def choose_torch_device --- invokeai/app/invocations/latent.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py index 2800624235..6ec62ff117 100644 --- a/invokeai/app/invocations/latent.py +++ b/invokeai/app/invocations/latent.py @@ -38,6 +38,10 @@ from diffusers.models.attention_processor import ( XFormersAttnProcessor, ) +def choose_torch_device() -> str: + # Your device selection logic here + device = "cuda" if torch.cuda.is_available() else "cpu" + return device class LatentsField(BaseModel): """A latents field used for passing latents between invocations""" From 19d67b29e7fc10281e5964a71cd9d30c7a0bc71e Mon Sep 17 00:00:00 2001 From: mickr777 <115216705+mickr777@users.noreply.github.com> Date: Wed, 19 Jul 2023 15:20:40 +1000 Subject: [PATCH 2/3] Remove not needed text --- invokeai/app/invocations/latent.py | 1 - 1 file changed, 1 deletion(-) diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py index 6ec62ff117..28613b1a4d 100644 --- a/invokeai/app/invocations/latent.py +++ b/invokeai/app/invocations/latent.py @@ -39,7 +39,6 @@ from diffusers.models.attention_processor import ( ) def choose_torch_device() -> str: - # Your device selection logic here device = "cuda" if torch.cuda.is_available() else "cpu" return device 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 3/3] 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"""