add badge

This commit is contained in:
Phil Wang
2020-09-14 13:38:06 -07:00
parent 4bf28914bc
commit 8e5fb17063
+2 -2
View File
@@ -6,6 +6,8 @@ Implementation of <a href="https://arxiv.org/abs/2006.11239">Denoising Diffusion
<img src="./sample.png" width="500px"><img>
[![PyPI version](https://badge.fury.io/py/denoising-diffusion-pytorch.svg)](https://badge.fury.io/py/denoising-diffusion-pytorch)
## Install
```bash
@@ -73,8 +75,6 @@ trainer = Trainer(
trainer.train()
```
Todo: Command line tool for one-line training
## Citations
```bibtex