Views
No views yet
| Item | Description |
|---|---|
| Base model | runwayml/stable-diffusion-v1-5 |
| Fine-tuned for | Flat, clean, Notion-style illustrations |
| Format | .safetensors |
| Framework | 🤗 diffusers |
| Author | @choeeiden |
| Model file | otion_1759657289_last.safetensors |
1from diffusers import StableDiffusionPipeline
2from huggingface_hub import hf_hub_download
3import torch
4
5# Download model
6local_path = hf_hub_download(
7 repo_id="choeeiden/notion-style-illustration-models",
8 filename="weights/otion_1759657289_last.safetensors"
9)
10
11# Load pipeline
12pipe = StableDiffusionPipeline.from_single_file(
13 pretrained_model_link_or_path=local_path,
14 torch_dtype=torch.float16 # Use torch.float32 on CPU
15).to("cuda")
16
17# Generate image
18prompt = "a man reading a book while sitting on a stack of books, soft watercolor, minimal notion-style"
19image = pipe(prompt).images[0]
20image.save("output.png")