apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): defer first model load until first request

This commit is contained in:
Sean Sube 2023-01-08 00:08:50 -06:00
parent 4894e0ddd6
commit 0232c7180c
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/serve.py
View File

@ -151,8 +151,6 @@ def check_paths():
def load_models():
global available_models
available_models = [f.name for f in scandir(model_path) if f.is_dir()]
load_pipeline(OnnxStableDiffusionPipeline, get_model_path(available_models[0]), platform_providers.get(
default_platform), pipeline_schedulers.get(default_scheduler))
check_paths()