aws-neuron
/

SDXL-Lightning-4steps-neuronx

Model card Files Files and versions

Jingya HF Staff commited on Aug 27, 2024

Commit

92f9aad

·

verified ·

1 Parent(s): e26b2cf

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -2,7 +2,7 @@
 license: openrail++
 ---
-This checkkpoint is compiled on [ByteDance/SDXL-Lightning](https://huggingface.co/ByteDance/SDXL-Lightning) for AWS Inf2.
 ## Compilation
@@ -46,7 +46,7 @@ stable_diffusion.save_pretrained(save_directory)
 from optimum.neuron import NeuronStableDiffusionXLPipeline
 from diffusers import EulerDiscreteScheduler
-pipe = NeuronStableDiffusionXLPipeline.from_pretrained("Jingya/SDXL-Lightning-4steps-neuronx")
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
 pipe("A girl smiling", num_inference_steps=4, guidance_scale=0).images[0].save("output.png")

 license: openrail++
 ---
+This checkkpoint is compiled by [ByteDance/SDXL-Lightning](https://huggingface.co/ByteDance/SDXL-Lightning) for AWS Inf2.
 ## Compilation
 from optimum.neuron import NeuronStableDiffusionXLPipeline
 from diffusers import EulerDiscreteScheduler
+pipe = NeuronStableDiffusionXLPipeline.from_pretrained("aws-neuron/SDXL-Lightning-4steps-neuronx")
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
 pipe("A girl smiling", num_inference_steps=4, guidance_scale=0).images[0].save("output.png")