feat(api): enable long prompt weighting pipeline
This commit is contained in:
parent
9376de880e
commit
ddc6c9fb1a
|
@ -115,8 +115,9 @@ def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, schedu
|
||||||
pipe = last_pipeline_instance
|
pipe = last_pipeline_instance
|
||||||
else:
|
else:
|
||||||
print('loading different pipeline')
|
print('loading different pipeline')
|
||||||
pipe = pipeline.from_pretrained(
|
pipe = DiffusionPipeline.from_pretrained(
|
||||||
model,
|
model,
|
||||||
|
custom_pipeline='lpw_stable_diffusion_onnx',
|
||||||
provider=provider,
|
provider=provider,
|
||||||
safety_checker=None,
|
safety_checker=None,
|
||||||
scheduler=scheduler.from_pretrained(model, subfolder='scheduler')
|
scheduler=scheduler.from_pretrained(model, subfolder='scheduler')
|
||||||
|
|
Loading…
Reference in New Issue