wsj-hedcut-v1 / README.md
dmillar's picture
added images to README
4390485
|
raw
history blame
1.36 kB
metadata
license: creativeml-openrail-m
tags:
  - stable-diffusion
  - stable-diffusion-diffusers
  - text-to-image
inference: true

Wall Street Journal Hedcut Style for Stable Diffusion

Readers of the Wall Street Journal (WSJ) are familiar with the distinctive style used to create portaits of their writers and subjects. This is a fine-tuned stable diffusion model that can be used to create hedcut-styled images using the prompt wsj hedcut of <subject>. This model can also be used in a DreamBooth environment to train a face or other subject for custom and unique hedcuts.

#!pip install diffusers transformers scipy torch
from diffusers import StableDiffusionPipeline
import torch
model_id = "dmillar/wsj-hedcut-v1"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "wsj hedcut of a woman"
image = pipe(prompt).images[0]
image.save("./woman_hedcut.png")

Sample Images

Hepburn Mbappe Hanks Dog Cat