From f8f36e602c6182b7d66f51b4eb0a3dda9ac9fdf5 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Mar 2023 19:25:22 -0500 Subject: [PATCH] feat(api): load i18n labels from additional networks --- api/onnx_web/server/load.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py index 4803ffa2..c7d39a91 100644 --- a/api/onnx_web/server/load.py +++ b/api/onnx_web/server/load.py @@ -142,7 +142,7 @@ def load_extras(context: ServerContext): logger.debug("collecting strings from %s", file) merge(strings, data["strings"]) - for model_type in ["diffusion", "correction", "upscaling"]: + for model_type in ["diffusion", "correction", "upscaling", "networks"]: if model_type in data: for model in data[model_type]: if "label" in model: @@ -159,14 +159,28 @@ def load_extras(context: ServerContext): if "label" in inversion: inversion_name = inversion["name"] logger.debug( - "collecting label for inversion %s from %s", + "collecting label for Textual Inversion %s from %s", inversion_name, model_name, ) labels[ - f"inversion-{inversion_name}" + f"inversion.{inversion_name}" ] = inversion["label"] + if "loras" in model: + for lora in model["loras"]: + if "label" in lora: + lora_name = lora["name"] + logger.debug( + "collecting label for LoRA %s from %s", + lora_name, + model_name, + ) + labels[ + f"lora.{lora_name}" + ] = lora["label"] + + except Exception: logger.exception("error loading extras file")