1
0
Fork 0

fix(api): provide some necessary SDXL config

This commit is contained in:
Sean Sube 2023-11-25 12:29:17 -06:00
parent 6ecdae44a2
commit 02447f5fd6
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 1 deletions

View File

@ -218,7 +218,10 @@ def load_pipeline(
components["vae_decoder_session"],
components["text_encoder_session"],
components["unet_session"],
{}, # empty config
{
"force_zeros_for_empty_prompt": True,
"requires_aesthetics_score": False,
},
components["tokenizer"],
scheduler,
vae_encoder_session=components.get("vae_encoder_session", None),