mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
Update README.md
This commit is contained in:
committed by
GitHub
parent
6d4879c2e3
commit
2894be2a4f
23
README.md
23
README.md
@@ -37,3 +37,26 @@
|
||||
├── tests
|
||||
│ └── test_modeling_utils.py
|
||||
```
|
||||
|
||||
## Dummy Example
|
||||
```python
|
||||
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)
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user