File size: 862 Bytes
95ba608 a23c39b 740ee6a d71535c 494f2fa 740ee6a d71535c a23c39b f50533e 494f2fa 7ae7337 3591c2a 95ba608 3591c2a f50533e da36322 d71535c 3591c2a f50533e 7ae7337 d71535c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import torch
from diffusers import StableDiffusionPipeline
# Hugging Face API key
api_key = "your_huggingface_api_key"
# Load the Stable Diffusion pipeline with the API key
pipeline = StableDiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-2", # You can try using the newer and faster version of Stable Diffusion.
use_auth_token=api_key
)
# Move the pipeline to GPU if available
device = "cuda" if torch.cuda.is_available() else "cpu"
pipeline.to(device)
# Define a smaller resolution for faster generation
height = 512 # You can try 512 or 256 to make it faster
width = 512
# Generate an image
prompt = "A serene lake surrounded by mountains during sunset"
image = pipeline(prompt, height=height, width=width).images[0]
# Save the generated image
image.save("generated_image.png")
print("Image generated and saved successfully!")
|