fix(api): use diffusion model name without path when looking up hashes

ssube · Jun 27, 2023 · 0029943 · 0029943
1 parent 7cbb419
commit 0029943
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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,
     }