apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): enable long prompt weighting pipeline

This commit is contained in:
Sean Sube 2023-01-11 22:02:18 -06:00
parent 9376de880e
commit ddc6c9fb1a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/serve.py
View File

@ -115,8 +115,9 @@ def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, schedu
pipe = last_pipeline_instance
else:
print('loading different pipeline')
pipe = pipeline.from_pretrained(
pipe = DiffusionPipeline.from_pretrained(
model,
custom_pipeline='lpw_stable_diffusion_onnx',
provider=provider,
safety_checker=None,
scheduler=scheduler.from_pretrained(model, subfolder='scheduler')