metadata
license: creativeml-openrail-m
language:
- en
library_name: diffusers
tags:
- text-to-image
- stable-diffusion
base_model: stabilityai/stable-diffusion-2-1
graphic-art
Custom Stable Diffusion checkpoint (based on SD 2.1) for generating graphic-design related images.
⚠️ For best results set at least one side to
768px.
🧨 Diffusers
This model can be used just like any other Stable Diffusion model with HuggingFace pipelines, read the docs for more information.
import torch
from diffusers import StableDiffusionPipeline
model_id = "n6ai/sd-graphic-art"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "your prompt"
image = pipe(prompt).images[0]
image.save("./result.png")