|
--- |
|
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](https://huggingface.co/stabilityai/stable-diffusion-2-1)) for generating graphic-design related images. |
|
|
|
> Original model by [pmejna](https://civitai.com/user/pmejna) on [CivitAI](https://civitai.com/models/7884/graphic-art). |
|
|
|
> ⚠️ For best results set at least one side to `768px`. |
|
|
|
## 🧨 Diffusers |
|
|
|
This model can be used just like any other Stable Diffusion model with Hugging Face pipelines, |
|
[read the docs](https://huggingface.co/docs/diffusers/using-diffusers/loading) for more information. |
|
|
|
```python |
|
import torch |
|
from diffusers import StableDiffusionPipeline |
|
|
|
model_id = "n6ai/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") |
|
``` |