Flux implementation with SAEs
uv sync
git lfs clone https://huggingface.co/nev/taef1 somewhere/taef1
uv run python -m src.fae.t5
uv run python -m src.fae.sae_trainer
uv run python -m src.fae.server
- https://github.com/davisyoshida/qax -- quantization library for quantized T5 support.
- Author: offered a lot of help and advice for quantization with jax and
qax. Updated the library twice to support the needs of this project.
- Author: offered a lot of help and advice for quantization with jax and
- https://github.com/aredden -- gave advice for Flux accuracy issues
- https://github.com/markian-rybchuk -- tested the code, suggested improvements, helped with debugging