Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tidy(mm): use canonical capitalization for all model-related enums, classes

Browse Source 
For example, "Lora" -> "LoRA", "Vae" -> "VAE".
This commit is contained in:
psychedelicious
2024-03-05 17:37:17 +11:00
parent 4f9bb00275
commit 7c9128b253
13 changed files with 77 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/backend/model_manager/model_manager_fixtures.py
View File

@ -28,7 +28,7 @@ from invokeai.backend.model_manager.config import (
ModelSourceType,
ModelType,
ModelVariantType,
VaeDiffusersConfig,
VAEDiffusersConfig,
)
from invokeai.backend.model_manager.load import ModelCache, ModelConvertCache
from invokeai.backend.util.logging import InvokeAILogger
@ -162,13 +162,13 @@ def mm2_record_store(mm2_app_config: InvokeAIAppConfig) -> ModelRecordServiceBas
db = create_mock_sqlite_database(mm2_app_config, logger)
store = ModelRecordServiceSQL(db)
# add five simple config records to the database
config1 = VaeDiffusersConfig(
config1 = VAEDiffusersConfig(
key="test_config_1",
path="/tmp/foo1",
format=ModelFormat.Diffusers,
name="test2",
base=BaseModelType.StableDiffusion2,
type=ModelType.Vae,
type=ModelType.VAE,
hash="111222333444",
source="stabilityai/sdxl-vae",
source_type=ModelSourceType.HFRepoID,
@ -204,7 +204,7 @@ def mm2_record_store(mm2_app_config: InvokeAIAppConfig) -> ModelRecordServiceBas
format=ModelFormat.Diffusers,
name="test4",
base=BaseModelType.StableDiffusionXL,
type=ModelType.Lora,
type=ModelType.LoRA,
hash="111222333444",
source="author4/model4",
source_type=ModelSourceType.HFRepoID,
@ -215,7 +215,7 @@ def mm2_record_store(mm2_app_config: InvokeAIAppConfig) -> ModelRecordServiceBas
format=ModelFormat.Diffusers,
name="test5",
base=BaseModelType.StableDiffusion1,
type=ModelType.Lora,
type=ModelType.LoRA,
hash="111222333444",
source="author4/model5",
source_type=ModelSourceType.HFRepoID,

6
tests/backend/model_manager/util/test_hf_model_select.py
View File

@ -104,7 +104,7 @@ def sdxl_base_files() -> List[Path]:
],
),
(
ModelRepoVariant.DEFAULT,
ModelRepoVariant.Default,
[
"model_index.json",
"scheduler/scheduler_config.json",
@ -129,7 +129,7 @@ def sdxl_base_files() -> List[Path]:
],
),
(
ModelRepoVariant.OPENVINO,
ModelRepoVariant.OpenVINO,
[
"model_index.json",
"scheduler/scheduler_config.json",
@ -211,7 +211,7 @@ def sdxl_base_files() -> List[Path]:
],
),
(
ModelRepoVariant.FLAX,
ModelRepoVariant.Flax,
[
"model_index.json",
"scheduler/scheduler_config.json",