From 8269142a5466d23fd1e808705ccdad28da7ee8cd Mon Sep 17 00:00:00 2001 From: Jasmine Collins Date: Fri, 6 Oct 2023 08:45:33 -0700 Subject: [PATCH] Update diffusion/inference/inference_model.py Co-authored-by: Landan Seguin --- diffusion/inference/inference_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffusion/inference/inference_model.py b/diffusion/inference/inference_model.py index 86c9df51..4595d286 100644 --- a/diffusion/inference/inference_model.py +++ b/diffusion/inference/inference_model.py @@ -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