This commit is contained in:
Lincoln Stein 2023-10-12 21:23:29 -04:00
parent 89db8c83c2
commit 52274087f3
2 changed files with 8 additions and 0 deletions

View File

@ -152,6 +152,7 @@ async def import_model(
) -> ImportModelResponse:
"""Add a model using its local path, repo_id, or remote URL. Model characteristics will be probed and configured automatically"""
location = location.strip()
items_to_import = {location}
prediction_types = {x.value: x for x in SchedulerPredictionType}
logger = ApiDependencies.invoker.services.logger

View File

@ -236,9 +236,16 @@ class ModelInstall(object):
if not models_installed:
models_installed = dict()
model_path_id_or_url = str(model_path_id_or_url).strip()
# A little hack to allow nested routines to retrieve info on the requested ID
self.current_id = model_path_id_or_url
path = Path(model_path_id_or_url)
# fix relative paths
if path.exists() and not path.is_absolute():
path = path.absolute() # make relative to current WD
# checkpoint file, or similar
if path.is_file():
models_installed.update({str(path): self._install_path(path)})