vqrotate
One-line Rotation Trick for VQ-VAEs.
Drop-in wrapper that attaches the ICLR 2025 Rotation Trick to any VQ layer.
Features
- Single line of code:
attach_rotation(quantizer) - Multi-head support (lucidrains compatible)
- Mixed precision (AMP) ready
Install
pip install git+https://github.com/develepo/vqrotate.git
1 maintainer
Release files for vqrotate 0.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| vqrotate-0.2.0.tar.gz | 5.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vqrotate-0.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 12.4 kB
Release files / vqrotate-0.2.0.tar.gz
| Download URL | vqrotate-0.2.0.tar.gz |
|---|---|
| Size | 5.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
d7e7c2ca0ca405191071412f985caf7365fa0d4ff86a0ced2b233672f74ecc5f
|
|
BLAKE2b-256 checksum How to use checksums |
137bfd2965b6232b142bc26e5b18cba005a33cdfc83b1c17728972178424e2cc
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
Release files / vqrotate-0.2.0-py3-none-any.whl
| Download URL | vqrotate-0.2.0-py3-none-any.whl |
|---|---|
| Size | 6.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
3ce915bfbba9645a459ee30e4fce6ee6c462818231b0a7f6ae4b6abce63911f6
|
|
BLAKE2b-256 checksum How to use checksums |
bf0ee5accfae1f75bad0f4da533a7e8531d02fc078894cb1aea3a9e7899e5188
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
