TrpFrog-Diffusion
Collection
Models for generating images of TrpFrog.
•
9 items
•
Updated
•
1
This is a Stable Diffusion model fine-tuned on the trpfrog concept with DreamBooth. It can be used by modifying the instance_prompt
: a photo of trpfrog
TrpFrog (@Trpfrog) is a famous Japanese Twitter comedian.
Below is a screenshot of the site he is creating.
This is a fine-tuned model of Stable Diffusion with the powerful help of Dreambooth. The model is made specifically for TrpFrog and faithfully reproduces its appearance.
First install the necessary packages.
pip install diffusers transformers accelerate scipy safetensors
The image can then be generated with the following script.
import torch
from diffusers import StableDiffusionPipeline
keyword = "trpfrog"
prompt = f"a photo of {keyword}"
model_id = "Prgckwb/trpfrog-diffusion"
pipe = StableDiffusionPipeline.from_pretrained(
model_id, torch_dtype=torch.float16
).to("cuda")
image = pipe(prompt).images[0]
image.save("trpfrog.jpg")