From 0029943c7d83eb89721992845e1b8e9941bec22f Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 27 Jun 2023 07:09:40 -0500 Subject: [PATCH] fix(api): use diffusion model name without path when looking up hashes --- api/onnx_web/output.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/output.py b/api/onnx_web/output.py index a6eb0a3a7..732e41117 100644 --- a/api/onnx_web/output.py +++ b/api/onnx_web/output.py @@ -92,8 +92,10 @@ def str_params( inversions: List[Tuple[str, float]] = None, loras: List[Tuple[str, float]] = None, ) -> str: - model_hash = get_extra_hashes().get(params.model, "unknown") model_name = path.basename(path.normpath(params.model)) + logger.debug("getting model hash for %s", model_name) + + model_hash = get_extra_hashes().get(model_name, "unknown") hash_map = { model_name: model_hash, }