From 33eebacc0c8edc4307d44e490cba3b12e71c4b24 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 16 Jan 2023 20:37:59 -0600 Subject: [PATCH] lint(api): return models in order --- api/onnx_web/serve.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index b11b94f4..867ec307 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -218,15 +218,15 @@ def load_models(context: ServerContext): path.join(context.model_path, 'diffusion-*'))] diffusion_models.extend([ get_model_name(f) for f in glob(path.join(context.model_path, 'stable-diffusion-*'))]) - diffusion_models = list(set(diffusion_models)) + diffusion_models = list(set(diffusion_models)).sort() correction_models = [ get_model_name(f) for f in glob(path.join(context.model_path, 'correction-*'))] - correction_models = list(set(correction_models)) + correction_models = list(set(correction_models)).sort() upscaling_models = [ get_model_name(f) for f in glob(path.join(context.model_path, 'upscaling-*'))] - upscaling_models = list(set(upscaling_models)) + upscaling_models = list(set(upscaling_models)).sort() def load_params(context: ServerContext):