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.2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 99d3a9e7ebfc4d237db85b376799f7e508ca0fbb950c61901104cb0a3990506c |
|
MD5 | 5bb777e726a5757852d6a7b39724ff15 |
|
BLAKE2b-256 | 3502ec793d8f96b8760b7d3cc4410344ea2db65231b56b07b37c21ad5d158de4 |
Close
Hashes for autoregressive_diffusion_pytorch-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 540784d253ceaa460225df7274849c7027d8f6d25fb56c588b200ec6b5e8b5c1 |
|
MD5 | 08523bde7553e67c89202f8fcbba5733 |
|
BLAKE2b-256 | cf63d78592a1ff2c35de65cec01fcf7e550caf487c0309cd8709971d9f643f06 |