Code

Vae encode example with test#294

Code

Merged

aviator19941 merged 7 commits intomainnod-ai/SHARK-ModelDev:mainfrom vae_encodenod-ai/SHARK-ModelDev:vae_encodeCopy head branch name to clipboardDec 27, 2023

+181-65

Commits

Commits on Dec 27, 2023

• [WIP] tracing_mode="symbolic" fails for VAE encode

  aviator19941

  committed
  044844f

  Copy full SHA for 044844f

• [WIP] Update VAE encode accuracy after diffusers change

  aviator19941

  committed
  13b682c

  Copy full SHA for 13b682c

• [WIP] Add temp decomp for randn.generator to bypass VAE encode error

  aviator19941

  committed
  0694390

  Copy full SHA for 0694390

• Update sd_test to include vae encode

  aviator19941

  committed
  7ae597c

  Copy full SHA for 7ae597c

• Fix black formatting and keep decomps in separate file

  aviator19941

  committed
  669b145

  Copy full SHA for 669b145

• Add get_decompositions import for passes.py

  aviator19941

  committed
  1ba10b7

  Copy full SHA for 1ba10b7

• Add expected torch stable version

  aviator19941

  committed
  f97b1ea

  Copy full SHA for f97b1ea