from diffusers import StableDiffusionPipeline
import torch
# Load pre-trained Stable Diffusion model (you'll need an access token from Hugging Face)
pipe = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
torch_dtype=torch.float16,
revision="fp16",
use_auth_token="your_huggingface_token"
).to("cuda")
# ...
Weight: 1
This is pretty good:)