Update README.md
Browse files
README.md
CHANGED
|
@@ -49,7 +49,17 @@ TangoFlux consists of FluxTransformer blocks which are Diffusion Transformer (Di
|
|
| 49 |
|
| 50 |
|
| 51 |
## Getting Started
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
## Citation
|
| 54 |
|
| 55 |
```bibtex
|
|
|
|
| 49 |
|
| 50 |
|
| 51 |
## Getting Started
|
| 52 |
+
Download TangoFlux from our github https://github.com/declare-lab/TangoFlux
|
| 53 |
+
```python
|
| 54 |
+
import torchaudio
|
| 55 |
+
from tangoflux import TangoFluxInference
|
| 56 |
+
from IPython.display import Audio
|
| 57 |
|
| 58 |
+
model = TangoFluxInference(name='declare-lab/TangoFlux')
|
| 59 |
+
audio = model.generate('Hammer slowly hitting the wooden table', steps=50, duration=10)
|
| 60 |
+
|
| 61 |
+
Audio(data=audio, rate=44100)
|
| 62 |
+
```
|
| 63 |
## Citation
|
| 64 |
|
| 65 |
```bibtex
|