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_rotator(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.3.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.3.0.tar.gz | 5.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vqrotate-0.3.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 11.9 kB
Release files / vqrotate-0.3.0.tar.gz
| Download URL | vqrotate-0.3.0.tar.gz |
|---|---|
| Size | 5.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
911857e4ba5532b1262e0eeb314c5603506d5a9a00c8d0f041c785f9855647b0
|
|
BLAKE2b-256 checksum How to use checksums |
a29aa242f90bd85ab21c0a3c601eafa9f76b6f07d37db0d7706f320ec9f81f4d
|
| 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.3.0-py3-none-any.whl
| Download URL | vqrotate-0.3.0-py3-none-any.whl |
|---|---|
| Size | 6.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
54e5581940d4585d15ed8c4c447c55c0b1847f5d1587f63db9a97807f15f3fae
|
|
BLAKE2b-256 checksum How to use checksums |
a9f0fe136ca7680efa9603e316eb06668e45f09bc871cc1353c2691ae6867f65
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
