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):