Downloads · 30 days
0
small-sd-dreambooth-lora-dog
takuoko/small-sd-dreambooth-lora-dog
small-sd-dreambooth-lora-dog is a machine learning model from takuoko. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
This is the example model of this PR. The training is based on DiffEngine, the open-source toolbox for training state-of-the-art Diffusion Models with diffusers and mmengine.
Access
Public
Updated Oct 24, 2023
Repo size
4.1 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin4.1 MB · 70%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Aug 5, 2023
- Updated
- Oct 24, 2023
- SHA
- 7b31279c
Model card
From the Hugging Face model README
Introduction
This is the example model of this PR. The training is based on DiffEngine, the open-source toolbox for training state-of-the-art Diffusion Models with diffusers and mmengine.
Dataset
I used diffusers/dog-example.
Inference
import torch
from diffusers import DiffusionPipeline
checkpoint = 'takuoko/small-sd-dreambooth-lora-dog'
prompt = 'A photo of sks dog in a bucket'
pipe = DiffusionPipeline.from_pretrained(
'segmind/small-sd', torch_dtype=torch.float16)
pipe.to('cuda')
pipe.load_lora_weights(checkpoint, weight_name='pytorch_lora_weights.bin')
image = pipe(
prompt,
num_inference_steps=50,
).images[0]
image.save('demo.png')
Example result
prompt = 'A photo of sks dog in a bucket'
- License
- apache-2.0
- Created
- Aug 5, 2023
- Updated
- Oct 24, 2023