Downloads · 30 days
8
4% of all-time downloads
ddpm-dummpy-test
valhalla/ddpm-dummpy-test
ddpm-dummpy-test is a machine learning model from valhalla. 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 transformers.
All-time downloads
206
Public
Repo size
225 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt225 MB · 100%
At a glance
- Library
- transformers
- Access
- Public
- Created
- Jun 9, 2022
- Updated
- Jun 9, 2022
- SHA
- 75a54908
Model card
From the Hugging Face model README
Dummy diffusion model following architecture of https://github.com/lucidrains/denoising-diffusion-pytorch
Run the model as follows:
from diffusers import UNetModel, GaussianDiffusion
import torch
# 1. Load model
unet = UNetModel.from_pretrained("fusing/ddpm_dummy")
# 2. Do one denoising step with model
batch_size, num_channels, height, width = 1, 3, 32, 32
dummy_noise = torch.ones((batch_size, num_channels, height, width))
time_step = torch.tensor([10])
image = unet(dummy_noise, time_step)
# 3. Load sampler
sampler = GaussianDiffusion.from_config("fusing/ddpm_dummy")
# 4. Sample image from sampler passing the model
image = sampler.sample(model, batch_size=1)
print(image)
- Library
- transformers
- Created
- Jun 9, 2022
- Updated
- Jun 9, 2022