Fix problem of all installed models being assigned "<NOKEY>"

- Also fix redundant scanning of models directory at startup.
This commit is contained in:
Lincoln Stein
2024-02-29 21:31:53 -05:00
committed by Kent Keirsey
parent 26d4d93e64
commit 0832e1818e
2 changed files with 6 additions and 5 deletions

View File

@ -188,7 +188,7 @@ class ModelProbe(object):
and fields["prediction_type"] == SchedulerPredictionType.VPrediction
)
model_info = ModelConfigFactory.make_config(fields, key=fields.get("key", None))
model_info = ModelConfigFactory.make_config(fields) # , key=fields.get("key", None))
return model_info
@classmethod