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

added textual inversion and lora loaders

Browse Source
This commit is contained in:
Lincoln Stein
2024-02-04 23:18:00 -05:00
committed by psychedelicious
parent 67eb715093
commit 0d3addc69b
16 changed files with 701 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/backend/model_manager/load/load_default.py
View File

@ -10,11 +10,17 @@ from diffusers import ModelMixin
from diffusers.configuration_utils import ConfigMixin
from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.backend.model_manager import AnyModel, AnyModelConfig, InvalidModelConfigException, ModelRepoVariant, SubModelType
from invokeai.backend.model_manager import (
AnyModel,
AnyModelConfig,
InvalidModelConfigException,
ModelRepoVariant,
SubModelType,
)
from invokeai.backend.model_manager.load.convert_cache import ModelConvertCacheBase
from invokeai.backend.model_manager.load.load_base import LoadedModel, ModelLoaderBase
from invokeai.backend.model_manager.load.model_cache.model_cache_base import ModelCacheBase, ModelLockerBase
from invokeai.backend.model_manager.load.model_util import calc_model_size_by_fs, calc_model_size_by_data
from invokeai.backend.model_manager.load.model_util import calc_model_size_by_data, calc_model_size_by_fs
from invokeai.backend.model_manager.load.optimizations import skip_torch_weight_init
from invokeai.backend.util.devices import choose_torch_device, torch_dtype
@ -160,4 +166,3 @@ class ModelLoader(ModelLoaderBase):
submodel_type: Optional[SubModelType] = None,
) -> AnyModel:
raise NotImplementedError