Skip to content

Commit

Permalink
fix(api): provide some necessary SDXL config
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Nov 25, 2023
1 parent 6ecdae4 commit 02447f5
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion api/onnx_web/diffusers/load.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand Down

0 comments on commit 02447f5

Please sign in to comment.