Code

[Flux] Add batched inference#1227

Code

Open

CarlosGomes98 wants to merge 6 commits intopytorch:mainpytorch/torchtitan:mainfrom

CarlosGomes98:cgomes/batched_inferenceCarlosGomes98/torchtitan:cgomes/batched_inferenceCopy head branch name to clipboard

Commits

Commits on May 27, 2025

• add batched inference

  

  Carlos Gomes

  committed
  929fd66

  Copy full SHA for 929fd66

• simplify sampling.py structure

  

  Carlos Gomes

  committed
  afabe1a

  Copy full SHA for afabe1a

• remove unused function from sampling

  

  Carlos Gomes

  committed
  ee9adbf

  Copy full SHA for ee9adbf

• Add prompt information to metadata, save all images

  

  Carlos Gomes

  committed
  56d5ece

  Copy full SHA for 56d5ece

• remove wrong comment

  

  Carlos Gomes

  committed
  e6ac626

  Copy full SHA for e6ac626

• formatting changes

  

  Carlos Gomes

  committed
  eccbbe3

  Copy full SHA for eccbbe3