Code

Add a shortfin pipeline for flux#876

Code

Merged

KyleHerndon merged 32 commits intomainnod-ai/shark-ai:mainfrom kyle-fluxnod-ai/shark-ai:kyle-fluxCopy head branch name to clipboardFeb 17, 2025

+4,169-182

Commits

Commits on Jan 20, 2025

• Create the Flux Pipeline and tests

  KyleHerndon

  committed
  6f148c2

  Copy full SHA for 6f148c2

• More updates

  KyleHerndon

  committed
  af6e9d7

  Copy full SHA for af6e9d7

• Add dynamo exports for flux.1

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  b58cb11

  Copy full SHA for b58cb11

• Add README

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  0dff3e5

  Copy full SHA for 0dff3e5

• Remove commented code.

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  ec92d2f

  Copy full SHA for ec92d2f

• Initial commit of flux server.

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  58fa0f1

  Copy full SHA for 58fa0f1

• Remove breakpoints.

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  c75adda

  Copy full SHA for c75adda

• Fixup model exports and service.

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  a45383c

  Copy full SHA for a45383c

• dtype flexibility for flux.

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  a7592c0

  Copy full SHA for a7592c0

• Add onnx export of mmdit and rename dynamo_exports dir to torch_exports

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  62d552d

  Copy full SHA for 62d552d

• Update README

  

  eagarvey-amd

  authored and

  KyleHerndon

  committed
  cfd3519

  Copy full SHA for cfd3519

Commits on Jan 28, 2025

• Updates that get everything but VAE working

  KyleHerndon

  committed
  0e65537

  Copy full SHA for 0e65537

• Simplify export code to one file

  KyleHerndon

  committed
  04c1a0e

  Copy full SHA for 04c1a0e

Commits on Jan 29, 2025

• Refactor

  KyleHerndon

  committed
  f6e4a81

  Copy full SHA for f6e4a81

Commits on Jan 31, 2025

• Fix up everything. BF16 still failing though

  KyleHerndon

  committed
  7fe1c1d

  Copy full SHA for 7fe1c1d

Commits on Feb 6, 2025

• Fix grayscale bug and allow mixing different types of decoders with the denoiser

  KyleHerndon

  committed
  53bf17a

  Copy full SHA for 53bf17a

• Latest fixes

  KyleHerndon

  committed
  63d46f4

  Copy full SHA for 63d46f4

Commits on Feb 10, 2025

• Typo fixes and style correction

  KyleHerndon

  committed
  b2aa5c8

  Copy full SHA for b2aa5c8

• Additional cleanup

  KyleHerndon

  committed
  98870e3

  Copy full SHA for 98870e3

• More cleanup

  KyleHerndon

  committed
  88887dc

  Copy full SHA for 88887dc

Commits on Feb 12, 2025

• Many refactors

  KyleHerndon

  committed
  fae7ccd

  Copy full SHA for fae7ccd

• Refactor constants to config

  KyleHerndon

  committed
  099839b

  Copy full SHA for 099839b

• run pre-commit

  KyleHerndon

  committed
  778ea11

  Copy full SHA for 778ea11

• Pre-commit in sharktank too

  KyleHerndon

  committed
  e631a5a

  Copy full SHA for e631a5a

Commits on Feb 16, 2025

• Fix typos

  KyleHerndon

  committed
  0ae68eb

  Copy full SHA for 0ae68eb

Commits on Feb 17, 2025

• Final cleanup

  KyleHerndon

  committed
  e31c43a

  Copy full SHA for e31c43a

• Merge branch 'main' into kyle-flux

  KyleHerndon

  authored
  9f8ff73

  Copy full SHA for 9f8ff73

• Run pre-commit, fix sd

  KyleHerndon

  committed
  ff7bbb7

  Copy full SHA for ff7bbb7

• SD fix

  KyleHerndon

  committed
  e76faa1

  Copy full SHA for e76faa1

• Missed some files

  KyleHerndon

  committed
  3ba3f60

  Copy full SHA for 3ba3f60

• Fix refactor

  KyleHerndon

  committed
  daf06a6

  Copy full SHA for daf06a6

• Fix needs_file_url in sd

  KyleHerndon

  committed
  ed01ac4

  Copy full SHA for ed01ac4