Autoregressive Diffusion - Pytorch
Project description
Autoregressive Diffusion - Pytorch (wip)
Implementation of the architecture behind Autoregressive Image Generation without Vector Quantization in Pytorch
You can discuss the paper temporarily here
Install
$ pip install autoregressive-diffusion-pytorch
Usage
import torch
from autoregressive_diffusion_pytorch import AutoregressiveDiffusion
model = AutoregressiveDiffusion(
dim = 512,
max_seq_len = 32
)
seq = torch.randn(3, 32, 512)
loss = model(seq)
loss.backward()
sampled = model.sample(batch_size = 3)
assert sampled.shape == seq.shape
Citations
@article{Li2024AutoregressiveIG,
title = {Autoregressive Image Generation without Vector Quantization},
author = {Tianhong Li and Yonglong Tian and He Li and Mingyang Deng and Kaiming He},
journal = {ArXiv},
year = {2024},
volume = {abs/2406.11838},
url = {https://api.semanticscholar.org/CorpusID:270560593}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for autoregressive_diffusion_pytorch-0.0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0a03ed67519e8e98b4bc75354bacbef594ce95ee313fff693e5b96057398070e |
|
MD5 | 91d5ed2e876a867aaf5c784137e436a2 |
|
BLAKE2b-256 | 9712fc9d1ed9f4ef0ab54031cc7c5f1dc208caf57ede72b2e1cf81c9d485918c |
Close
Hashes for autoregressive_diffusion_pytorch-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4fa307af86b471b7665fa3a313a6f5107956293c07bc18e90a311adcaa35a447 |
|
MD5 | afe097d8e521bc9d1ab1881469362f44 |
|
BLAKE2b-256 | 308374c2ca769c8f28bf65dd43ddebea9e585afd91153332dca3096554580e3a |