Skip to content

Commit 67e3d3f

Browse files
Copilotlstein
andcommitted
Add cpu_only field to encoder model configs
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>
1 parent ef68700 commit 67e3d3f

File tree

4 files changed

+4
-0
lines changed

4 files changed

+4
-0
lines changed

invokeai/backend/model_manager/configs/clip_embed.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ class CLIPEmbed_Diffusers_Config_Base(Diffusers_Config_Base):
4141
base: Literal[BaseModelType.Any] = Field(default=BaseModelType.Any)
4242
type: Literal[ModelType.CLIPEmbed] = Field(default=ModelType.CLIPEmbed)
4343
format: Literal[ModelFormat.Diffusers] = Field(default=ModelFormat.Diffusers)
44+
cpu_only: bool | None = Field(default=None, description="Whether this model should run on CPU only")
4445

4546
@classmethod
4647
def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -> Self:

invokeai/backend/model_manager/configs/clip_vision.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ class CLIPVision_Diffusers_Config(Diffusers_Config_Base, Config_Base):
2828
base: Literal[BaseModelType.Any] = Field(default=BaseModelType.Any)
2929
type: Literal[ModelType.CLIPVision] = Field(default=ModelType.CLIPVision)
3030
format: Literal[ModelFormat.Diffusers] = Field(default=ModelFormat.Diffusers)
31+
cpu_only: bool | None = Field(default=None, description="Whether this model should run on CPU only")
3132

3233
@classmethod
3334
def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -> Self:

invokeai/backend/model_manager/configs/llava_onevision.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ class LlavaOnevision_Diffusers_Config(Diffusers_Config_Base, Config_Base):
2525

2626
type: Literal[ModelType.LlavaOnevision] = Field(default=ModelType.LlavaOnevision)
2727
base: Literal[BaseModelType.Any] = Field(default=BaseModelType.Any)
28+
cpu_only: bool | None = Field(default=None, description="Whether this model should run on CPU only")
2829

2930
@classmethod
3031
def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -> Self:

invokeai/backend/model_manager/configs/siglip.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ class SigLIP_Diffusers_Config(Diffusers_Config_Base, Config_Base):
2727
type: Literal[ModelType.SigLIP] = Field(default=ModelType.SigLIP)
2828
format: Literal[ModelFormat.Diffusers] = Field(default=ModelFormat.Diffusers)
2929
base: Literal[BaseModelType.Any] = Field(default=BaseModelType.Any)
30+
cpu_only: bool | None = Field(default=None, description="Whether this model should run on CPU only")
3031

3132
@classmethod
3233
def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -> Self:

0 commit comments

Comments
 (0)