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

proxy SDModelInfo so that it can be used directly as context

Browse Source
This commit is contained in:
Lincoln Stein 2023-05-13 16:29:18 -04:00
parent b23c9f1da5
commit d8b1f29066
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/model_cache.py
View File

@ -68,7 +68,7 @@ class SDModelType(Enum):
# distinguish them by class # distinguish them by class
lora=LoraType lora=LoraType
textual_inversion=TIType textual_inversion=TIType
class ModelStatus(Enum): class ModelStatus(Enum):
unknown='unknown' unknown='unknown'
not_loaded='not loaded' not_loaded='not loaded'

6
invokeai/backend/model_management/model_manager.py
View File

@ -176,6 +176,12 @@ class SDModelInfo():
revision: str = None revision: str = None
_cache: ModelCache = None _cache: ModelCache = None
def __enter__(self):
return self.context.__enter__()
def __exit__(self,*args, **kwargs):
self.context.__exit__(*args, **kwargs)
@property @property
def status(self)->ModelStatus: def status(self)->ModelStatus:
'''Return load status of this model as a model_cache.ModelStatus enum''' '''Return load status of this model as a model_cache.ModelStatus enum'''