Skip to content

Commit

Permalink
Update diffusion/inference/inference_model.py
Browse files Browse the repository at this point in the history
Co-authored-by: Landan Seguin <[email protected]>
  • Loading branch information
jazcollins and Landanjs authored Oct 6, 2023
1 parent 1c76f2a commit 8269142
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion diffusion/inference/inference_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ class StableDiffusionXLInference():
"""Inference endpoint class for Stable Diffusion XL.
Args:
model_name (str, optional): Name of the model to load. Default: 'stabilityai/stable-diffusion-xl-base-1.0'.
model_name (str): Name of the model to load. Default: 'stabilityai/stable-diffusion-xl-base-1.0'.
unet_model_name (str): Name of the UNet model to load. Default: 'stabilityai/stable-diffusion-xl-base-1.0'.
vae_model_name (str): Name of the VAE model to load. Defaults to
'madebyollin/sdxl-vae-fp16-fix' as the official VAE checkpoint (from
Expand Down

0 comments on commit 8269142

Please sign in to comment.