apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

list models correctly

This commit is contained in:
Sean Sube 2023-01-16 20:17:01 -06:00
parent 0273dea2a6
commit 23a9d5afb7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/serve.py
View File

@ -209,7 +209,7 @@ def load_models(context: ServerContext):
global upscaling_models global upscaling_models
diffusion_models = glob(path.join(context.model_path, 'diffusion-*')) diffusion_models = glob(path.join(context.model_path, 'diffusion-*'))
diffusion_models.append(glob(path.join(context.model_path, 'stable-diffusion-*'))) diffusion_models.extend(glob(path.join(context.model_path, 'stable-diffusion-*')))
correction_models = glob(path.join(context.model_path, 'correction-*')) correction_models = glob(path.join(context.model_path, 'correction-*'))
upscaling_models = glob(path.join(context.model_path, 'upscaling-*')) upscaling_models = glob(path.join(context.model_path, 'upscaling-*'))