Downloads · 30 days
90
8% of all-time downloads
tiny-random-unet2d
adams-story/tiny-random-unet2d
tiny-random-unet2d is a machine learning model from adams-story. 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.
This is the code used to create this model
All-time downloads
1.1K
Public
Parameters
10.7K
174 KB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors86.7 KB · 96%
At a glance
- Library
- diffusers
- Access
- Public
- Created
- Feb 22, 2024
- Updated
- Feb 23, 2024
- SHA
- 3b892aa4
Model card
From the Hugging Face model README
This is the code used to create this model
import torch
import diffusers
model = diffusers.UNet2DConditionModel(
block_out_channels=(4, 4, 4),
down_block_types=('CrossAttnDownBlock2D', 'CrossAttnDownBlock2D', 'CrossAttnDownBlock2D'),
up_block_types=('CrossAttnUpBlock2D', 'CrossAttnUpBlock2D', 'CrossAttnUpBlock2D'),
norm_num_groups=2,
cross_attention_dim=2,
layers_per_block=1,
attention_head_dim=2,
addition_embed_type_num_heads=2,
)
# noisy latent
x = torch.randn(7,4,33,33)
# timestep
t = torch.Tensor([1.0])
# conditioning embed
z = torch.randn(7, 4, 2)
# denoised latent
y = model(x, t, z)
- Library
- diffusers
- Created
- Feb 22, 2024
- Updated
- Feb 23, 2024