1
0
Fork 0

fix(api): abandon pretrained loader entirely to fix SD upscaling

This commit is contained in:
Sean Sube 2023-11-26 14:13:45 -06:00
parent 75f1a2cead
commit fa38b474f0
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 24 additions and 9 deletions

View File

@ -230,15 +230,30 @@ def load_pipeline(
)
else:
logger.debug(
"loading pretrained SD pipeline for %s", pipeline_class.__name__
"assembling SD pipeline for %s", pipeline_class.__name__
)
pipe = pipeline_class.from_pretrained(
model,
provider=device.ort_provider(),
sess_options=device.sess_options(),
safety_checker=None,
torch_dtype=torch_dtype,
**components,
if pipeline_class == OnnxStableDiffusionUpscalePipeline:
# upscale uses a single VAE
pipe = pipeline_class(
components["vae"],
components["text_encoder"],
components["tokenizer"],
components["unet"],
scheduler,
scheduler,
)
else:
pipe = pipeline_class(
components["vae_encoder"],
components["vae_decoder"],
components["text_encoder"],
components["tokenizer"],
components["unet"],
scheduler,
None,
None,
requires_safety_checker=False,
)
if not server.show_progress: