show how to load SDXL in README.md

chengzeyi · Nov 4, 2023 · 04c6802 · 04c6802
1 parent 746db79
commit 04c6802
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -104,8 +104,10 @@ from sfast.compilers.stable_diffusion_pipeline_compiler import (compile,
                                                                 )
 
 def load_model():
+    # you can change to StableDiffusionXLPipeline to load SDXL model
     model = StableDiffusionPipeline.from_pretrained(
         'runwayml/stable-diffusion-v1-5', torch_dtype=torch.float16)
+
     model.scheduler = EulerAncestralDiscreteScheduler.from_config(
         model.scheduler.config)
     model.safety_checker = None