|
|
|
--- |
|
license: creativeml-openrail-m |
|
tags: |
|
- pytorch |
|
- diffusers |
|
- stable-diffusion |
|
- text-to-image |
|
- diffusion-models-class |
|
- dreambooth-hackathon |
|
- wildcard |
|
datasets: |
|
- TrpFrog/trpfrog-icons |
|
widget: |
|
- text: a photo of trpfrog |
|
--- |
|
|
|
# DreamBooth model for the trpfrog concept trained by Prgckwb on the TrpFrog/trpfrog-icons dataset. |
|
|
|
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** |
|
 |
|
|
|
## Description |
|
TrpFrog ([@Trpfrog](https://twitter.com/trpfrog?s=21&t=GcDGy74adYhOBYxX9HloOg)) 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. |
|
|
|
|
|
|
|
## Usage |
|
First install the necessary packages. |
|
|
|
```bash |
|
pip install diffusers transformers accelerate scipy safetensors |
|
``` |
|
|
|
The image can then be generated with the following script. |
|
|
|
```python |
|
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") |
|
``` |
|
|
|
## Generated Images |
|
**a photo of trpfrog in spring** |
|
 |
|
|
|
**a photo of trpfrog in summer** |
|
 |
|
|
|
**a photo of trpfrog in autumn** |
|
 |
|
|
|
**a photo of trpfrog in winter** |
|
 |
|
|
|
**a photo of trpfrog on fire** |
|
 |
|
|
|
**a photo of trpfrog growing shiitake mushrooms** |
|
 |
