Code

TRT decoder#45

Code

Open

technillogue wants to merge 5 commits intomainreplicate/cog-flux:mainfrom syl/trt-decoder-and-unrollreplicate/cog-flux:syl/trt-decoder-and-unrollCopy head branch name to clipboard

+116-56

Commits

Commits on Nov 8, 2024

• remove some einops usage

  technillogue

  committed
  3bf7029

  Copy full SHA for 3bf7029

• formatting, initial changes, types, inlining, and initial attempt

  technillogue

  committed
  fd1546d

  Copy full SHA for fd1546d

• use nn.Sequential to remove python control flow from autoencoder up/downsampling

  technillogue

  committed
  9db707f

  Copy full SHA for 9db707f

• try to keep the same state_dict structure

  Show description for 29aa8c6
  technillogue

  committed
  29aa8c6

  Copy full SHA for 29aa8c6

• conditionally download and use decoder engine

  technillogue

  committed
  c982583

  Copy full SHA for c982583