apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): join globs to avoid py 3.10-only args

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

8
api/onnx_web/serve.py
View File

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