Downloads · 30 days
5
24% of all-time downloads
ddpm-butterflies-128
hjjeon/ddpm-butterflies-128
ddpm-butterflies-128 is a machine learning model from hjjeon. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for diffusers. The card lists the license as apache-2.0.
should probably proofread and complete it, then remove this comment. --
All-time downloads
21
Public
Repo size
2.3 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin455 MB · 99%
At a glance
- Library
- diffusers
- License
- apache-2.0
- Access
- Public
- Created
- Dec 7, 2022
- Updated
- Dec 9, 2022
- SHA
- 78ee1f0c
Model card
From the Hugging Face model README
<!-- This model card has been generated automatically according to the information the training script had access to. You
should probably proofread and complete it, then remove this comment. -->
ddpm-butterflies-128
Model description
This diffusion model is trained with the 🤗 Diffusers library
on the huggan/smithsonian_butterflies_subset dataset.
Intended uses & limitations
How to use
from diffusers import DDPMPipeline
model_id = "hjjeon/ddpm-butterflies-128"
# load model and scheduler
pipeline = DDPMPipeline.from_pretrained(model_id)
# run pipeline in inference
image = pipeline()["sample"]
# save image
image[0].save("butterfly.png")
Limitations and bias
[TODO: provide examples of latent issues and potential remediations]
Training data
[TODO: describe the data used to train the model]
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0001
- train_batch_size: 16
- eval_batch_size: 16
- gradient_accumulation_steps: 1
- optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None
- lr_scheduler: None
- lr_warmup_steps: 500
- ema_inv_gamma: None
- ema_inv_gamma: None
- ema_inv_gamma: None
- mixed_precision: fp16
Training results
- Library
- diffusers
- License
- apache-2.0
- Languages
- en
- Created
- Dec 7, 2022
- Updated
- Dec 9, 2022