apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): return models in order

This commit is contained in:
Sean Sube 2023-01-16 20:37:59 -06:00
parent c0ca7cf62f
commit 33eebacc0c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/serve.py
View File

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