Code

torch.compile ae.decode#25

Code

Open

yorickvP wants to merge 4 commits intomainreplicate/cog-flux:mainfrom yorickvp/torch-compile-vaereplicate/cog-flux:yorickvp/torch-compile-vaeCopy head branch name to clipboard

+80-22

Commits

Commits on Oct 10, 2024

• torch.compile ae.decode

  yorickvP

  committed
  08fc060

  Copy full SHA for 08fc060

• Fix ruff errors

  yorickvP

  committed
  4dd56f6

  Copy full SHA for 4dd56f6

• Add compile_ae argument to base_setup

  yorickvP

  committed
  f676e86

  Copy full SHA for f676e86

• fp8/pipeline: add some profiler annotation for prepare/denoise/vae

  yorickvP

  committed
  0039a42

  Copy full SHA for 0039a42