| license: creativeml-openrail-m | |
| tags: | |
| - pytorch | |
| - diffusers | |
| - Stable-Diffusion-Model | |
| - diffusion-models-class | |
| This model is a diffusion model for text to image generation. | |
| ## Usage | |
| ```python | |
| from diffusers import StableDiffusionPipeline | |
| import torch | |
| pipeline = StableDiffusionPipeline.from_pretrained('MADZaMeR/stable_diffusion', torch_dtype=torch.float16 | |
| ).to("cuda" if torch.cuda.is_available() else "cpu") | |
| prompt = "a photograph of an astronaut riding a horse" | |
| image = pipeline(prompt, num_inference_steps=50, guidance_scale=7.5).images[0] | |
| image | |
| ``` | |