GitXplorerGitXplorer
p

vit-vqgan

public
80 stars
11 forks
11 issues

Commits

List of commits on branch main.
Verified
15039d7fc03e8e1c29440a88e9f87b9e9c0da5ff

feat: support ROPE (#23)

bborisdayma committed 2 years ago
Verified
b80844987fa7a7239eda6bcebaa23f1c64d2c86e

feat: add scanning + sharding (#26)

bborisdayma committed 2 years ago
Verified
6dce733329541129f0d60cdce2487a340e726abf

feat: update shampoo (#25)

bborisdayma committed 2 years ago
Verified
6348094461bc7594a5aebe4aa088d242b9513c1a

feat(data): faster reading + allow buckets (#24)

bborisdayma committed 2 years ago
Verified
abef041b9a1c0d875fc13026469148c34b79e1cf

feat: add stylegan loss (#15)

bborisdayma committed 2 years ago
Verified
038409f35c7a89dbb593e3301e1e999f8cf40a07

allow different number of layers in enc & dec (#19)

ppatil-suraj committed 2 years ago

README

The README file for this repository.

vit-vqgan

JAX implementation of ViT-VQGAN.

Acknowledgements

  • Jiahui Yu for his advice on the implementation based on the work on ViT-VQGAN and Parti.
  • Phil Wang for suggesting using convolutions after self-attention layers.
  • Katherine Crowson for suggesting using convolutions in feed-forward layers.