apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-03-18 21:48:51 -05:00
parent 645da3d1a6
commit 9a949806d3
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/onnx_web/server/load.py
View File

@ -154,7 +154,9 @@ def load_extras(context: ServerContext):
) )
if "type" in model: if "type" in model:
labels[f'{model["type"]}.{model_name}'] = model["label"] labels[f'{model["type"]}.{model_name}'] = model[
"label"
]
else: else:
labels[model_name] = model["label"] labels[model_name] = model["label"]
@ -180,10 +182,7 @@ def load_extras(context: ServerContext):
lora_name, lora_name,
model_name, model_name,
) )
labels[ labels[f"lora.{lora_name}"] = lora["label"]
f"lora.{lora_name}"
] = lora["label"]
except Exception: except Exception:
logger.exception("error loading extras file") logger.exception("error loading extras file")