Code

Flux Autoencoder#2098

Code

Merged

calvinpelletier merged 9 commits intopytorch:mainpytorch/torchtune:mainfrom calvinpelletier:flux_aecalvinpelletier/torchtune:flux_aeCopy head branch name to clipboardJan 8, 2025

+627

Commits

Commits on Nov 30, 2024

• flux autoencoder builder

  calvinpelletier

  committed
  7f9641a

  Copy full SHA for 7f9641a

• flux autoencoder

  calvinpelletier

  committed
  bfc93fc

  Copy full SHA for bfc93fc

• convert flux autoencoder weights

  calvinpelletier

  committed
  0e6611e

  Copy full SHA for 0e6611e

Commits on Dec 2, 2024

• flux autoencoder unit test

  calvinpelletier

  committed
  e7504b6

  Copy full SHA for e7504b6

• Merge remote-tracking branch 'origin/main' into flux_ae

  calvinpelletier

  committed
  aa8b751

  Copy full SHA for aa8b751

Commits on Jan 7, 2025

• Merge remote-tracking branch 'origin/main' into flux_ae

  calvinpelletier

  committed
  ddd2e7d

  Copy full SHA for ddd2e7d

• addressing comments

  calvinpelletier

  committed
  794b1bd

  Copy full SHA for 794b1bd

• fmt

  calvinpelletier

  committed
  346c51e

  Copy full SHA for 346c51e

Commits on Jan 8, 2025

• moving encoder/decoder construction to the model builder

  calvinpelletier

  committed
  c78f485

  Copy full SHA for c78f485