| This is a dummy model used for testing `NeuronStableDiffusionPipeline` of `optimum-neuron`. |
|
|
| ```python |
| from optimum.neuron import NeuronStableDiffusionPipeline |
| |
| model_id = "optimum/tiny-stable-diffusion-neuronx" |
| input_shapes = {"batch_size": 2, "sequence_length": 18, "num_channels": 4, "height": 64, "width": 64} |
| |
| stable_diffusion = NeuronStableDiffusionPipeline.from_pretrained(model_id, export=True, **input_shapes) |
| save_directory = "sd_neuron/" |
| stable_diffusion.save_pretrained(save_directory) |
| |
| |
| prompt = "sailing ship in storm by Leonardo da Vinci" |
| image = stable_diffusion(prompt).images[0] |
| ``` |
|
|
