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

Update doc string for import_local_model and remove access_token since it's only usable for local file paths

Browse Source
This commit is contained in:
Brandon Rising 2024-08-21 11:18:07 -04:00
parent e680cf76f6
commit d4872253a1
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
invokeai/app/services/shared/invocation_context.py
View File

@ -468,17 +468,24 @@ class ModelsInterface(InvocationContextInterface):
self, self,
model_path: Path, model_path: Path,
config: Optional[ModelRecordChanges] = None, config: Optional[ModelRecordChanges] = None,
access_token: Optional[str] = None,
inplace: Optional[bool] = False, inplace: Optional[bool] = False,
): ):
""" """
TODO: Fill out description of this method Import the model file located at the given local file path and return its ModelInstallJob.
This can be used to single-file models or directories.
Args:
model_path: A pathlib.Path object pointing to a model file or directory
config: Optional ModelRecordChanges to define manual probe overrides
inplace: Optional boolean to declare whether or not to install the model in the models dir
Returns:
ModelInstallJob object defining the install job to be used in tracking the job
""" """
if not model_path.exists(): if not model_path.exists():
raise Exception("Models provided to import_local_model must already exist on disk") raise Exception("Models provided to import_local_model must already exist on disk")
return self._services.model_manager.install.heuristic_import( return self._services.model_manager.install.heuristic_import(str(model_path), config=config, inplace=inplace)
str(model_path), config=config, access_token=access_token, inplace=inplace
)
def load_local_model( def load_local_model(
self, self,