Jingya HF staff commited on
Commit
14b22ed
1 Parent(s): 69ae377

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -3
README.md CHANGED
@@ -1,3 +1,15 @@
1
- ---
2
- license: openrail++
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: openrail++
3
+ ---
4
+
5
+ This checkkpoint is compiled on [ByteDance/SDXL-Lightning](https://huggingface.co/ByteDance/SDXL-Lightning) for AWS Inf2.
6
+
7
+ ```python
8
+ from optimum.neuron import NeuronStableDiffusionXLPipeline
9
+ from diffusers import EulerDiscreteScheduler
10
+
11
+ pipe = NeuronStableDiffusionXLPipeline.from_pretrained("Jingya/SDXL-Lightning-4steps-neuronx")
12
+ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
13
+ pipe("A girl smiling", num_inference_steps=4, guidance_scale=0).images[0].save("output.png")
14
+
15
+ ```