From b09feda47427a6a2d79c7006a5d3d6f1cab7bc8d Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 16 Jan 2023 20:58:08 -0600 Subject: [PATCH] fix(api): sort models without discarding --- api/onnx_web/serve.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 867ec307..4ea0dfcd 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -218,15 +218,18 @@ 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)).sort() + diffusion_models = list(set(diffusion_models)) + 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)).sort() + correction_models = list(set(correction_models)) + 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)).sort() + upscaling_models = list(set(upscaling_models)) + upscaling_models.sort() def load_params(context: ServerContext):