Skip to content

Run FLUX text encoders under Torchax instead of Flax - #483

Open
olehtika wants to merge 2 commits into
AI-Hypercomputer:mainfrom
olehtika:flux-torchax-text-encoders
Open

Run FLUX text encoders under Torchax instead of Flax#483
olehtika wants to merge 2 commits into
AI-Hypercomputer:mainfrom
olehtika:flux-torchax-text-encoders

Place Torchax text encoder buffers alongside params

a16fc1f
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Oct 22, 2026 in 2s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

✅ a16fc1f Author: @olehtika <olehtika​@users.noreply.github.com>

(Only the first commit for a unique contributor is listed.)

View more details on Google CLA