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

docs(mm): format docstrings for ModelSearch

Browse Source
This commit is contained in:
psychedelicious 2024-03-10 12:02:29 +11:00
parent 92b2e8186a
commit c47dbf7258
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
invokeai/backend/model_manager/search.py
View File

@ -4,19 +4,19 @@ Abstract base class and implementation for recursive directory search for models
Example usage: Example usage:
``` ```
from invokeai.backend.model_manager import ModelSearch, ModelProbe from invokeai.backend.model_manager import ModelSearch, ModelProbe
def find_main_models(model: Path) -> bool: def find_main_models(model: Path) -> bool:
info = ModelProbe.probe(model) info = ModelProbe.probe(model)
if info.model_type == 'main' and info.base_type == 'sd-1': if info.model_type == 'main' and info.base_type == 'sd-1':
return True return True
else: else:
return False return False
search = ModelSearch(on_model_found=report_it) search = ModelSearch(on_model_found=report_it)
found = search.search('/tmp/models') found = search.search('/tmp/models')
print(found) # list of matching model paths print(found) # list of matching model paths
print(search.stats) # search stats print(search.stats) # search stats
``` ```
""" """