fix(api): trim whitespace from model names because it breaks things (#376)

api/onnx_web/convert/diffusion/diffusion.py

@@ -275,7 +275,7 @@ def convert_diffusion_diffusers(
     """
     From https://github.com/huggingface/diffusers/blob/main/scripts/convert_stable_diffusion_checkpoint_to_onnx.py
     """
-    name = model.get("name")
+    name = str(model.get("name")).strip()
     source = model.get("source")
 
     # optional

api/onnx_web/convert/diffusion/diffusion_xl.py

@@ -25,7 +25,7 @@ def convert_diffusion_diffusers_xl(
     """
     From https://github.com/huggingface/diffusers/blob/main/scripts/convert_stable_diffusion_checkpoint_to_onnx.py
     """
-    name = model.get("name")
+    name = str(model.get("name")).strip()
     source = model.get("source")
     replace_vae = model.get("vae", None)
 

api/onnx_web/server/load.py

@@ -8,6 +8,7 @@ from typing import Any, Dict, List, Optional, Union
 import torch
 from jsonschema import ValidationError, validate
 
+from ..convert.utils import fix_diffusion_name
 from ..image import (  # mask filters; noise sources
     mask_filter_gaussian_multiply,
     mask_filter_gaussian_screen,
@@ -189,6 +190,9 @@ def load_extras(server: ServerContext):
                         for model in data[model_type]:
                             model_name = model["name"]
 
+                            if model_type == "diffusion":
+                                model_name = fix_diffusion_name(model_name)
+
                             if "hash" in model:
                                 logger.debug(
                                     "collecting hash for model %s from %s",