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.1
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.1.tar.gz | 5.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vqrotate-0.2.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 12.5 kB
Release files / vqrotate-0.2.1.tar.gz
| Download URL | vqrotate-0.2.1.tar.gz |
|---|---|
| Size | 5.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
b43370e31522b70832b70d9af434ed54c1bfb59b5953c58e817f77aff9d1e775
|
|
BLAKE2b-256 checksum How to use checksums |
89353d0955dd3ff445a06a597a3e4b5570b2ad4668c7f83f4d76aa843505a248
|
| 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.1-py3-none-any.whl
| Download URL | vqrotate-0.2.1-py3-none-any.whl |
|---|---|
| Size | 6.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
8b1a1f65f8c79ec99e7b07c2a08f910a4fcdab8a7efe1d621442d64251df4dc8
|
|
BLAKE2b-256 checksum How to use checksums |
8bc9676f34a510a6edd34bd66d5a404227a2ae5fd2bdefa7efe76d559145208f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
