fix "unknown model type" error when rebasing a model with API

- Add command-line model probing script for dev use
- Minor documentation tweak
This commit is contained in:
Lincoln Stein
2023-07-25 08:36:57 -04:00
parent 8c449c4756
commit 2aefa921fe
4 changed files with 27 additions and 6 deletions

View File

@ -391,6 +391,10 @@ class ModelManager(object):
base_model: BaseModelType,
model_type: ModelType,
) -> str:
# In 3.11, the behavior of (str,enum) when interpolated into a
# string has changed. The next two lines are defensive.
base_model = BaseModelType(base_model)
model_type = ModelType(model_type)
return f"{base_model.value}/{model_type.value}/{model_name}"
@classmethod