zelda-the-cat / README.md
Sander Land
Upload README.md with huggingface_hub
b852105
|
raw
history blame
1.13 kB
metadata
license: creativeml-openrail-m
tags:
  - pytorch
  - diffusers
  - stable-diffusion
  - text-to-image
  - diffusion-models-class
  - dreambooth-hackathon
  - animal
widget:
  - text: a photo of three zzelda cat attacking an astronaut in space

DreamBooth model for the zzelda concept trained by Sanderbaduk on dataset of cats.

This is a Stable Diffusion model fine-tuned on the zzelda concept with DreamBooth. It can be used by using the phrase 'zzelda cat' in a prompt.**

This model was created as part of the DreamBooth Hackathon 🔥. Visit the organisation page for instructions on how to take part!

Description

This is a Stable Diffusion model fine-tuned on images of my mum's cat Zelda for the animal theme.

To experiment a bit, I used a custom prompt for each image based on the file name. This was trained on CPU after encountering issues with CUDA, taking around 2 hours on 32 cores.

Usage

from diffusers import StableDiffusionPipeline

pipeline = StableDiffusionPipeline.from_pretrained('Sanderbaduk/zelda-the-cat')
image = pipeline().images[0]
image