Code

Add a shortfin pipeline for flux#876

Code

Open

KyleHerndon wants to merge 15 commits intomainnod-ai/shark-ai:mainfrom kyle-fluxnod-ai/shark-ai:kyle-fluxCopy head branch name to clipboard

+3,361-23

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