Code

Adding jax maskgit weights to muse#12

Code

Open

isamu-isozaki wants to merge 14 commits intomainhuggingface/open-muse:mainfrom jax_maskgit_loadinghuggingface/open-muse:jax_maskgit_loadingCopy head branch name to clipboard

+252-26

Commits

Commits on Mar 18, 2023

• init commit

  isamu-isozaki

  committed
  3605f5e

  Copy full SHA for 3605f5e

• Removed maskgit dependency+updated huggingface_hub required

  isamu-isozaki

  committed
  8f9b186

  Copy full SHA for 8f9b186

• Inference and maskgit weight loading fixed

  isamu-isozaki

  committed
  57b1880

  Copy full SHA for 57b1880

• Removed conv layer for transformer as there is no conv layers

  isamu-isozaki

  committed
  2f0050a

  Copy full SHA for 2f0050a

• Attempt adapting from maskgit-torch

  isamu-isozaki

  committed
  2447708

  Copy full SHA for 2447708

• Fixed eps and removed logging

  isamu-isozaki

  committed
  194bcab

  Copy full SHA for 194bcab

Commits on Mar 20, 2023

• Fixed setup

  isamu-isozaki

  committed
  04aaa97

  Copy full SHA for 04aaa97

Commits on Mar 26, 2023

• Fixed ordering to be the same as maskgit

  isamu-isozaki

  committed
  fc02eb5

  Copy full SHA for fc02eb5

Commits on Mar 27, 2023

• Fully made the jax to muse maskgit converter

  isamu-isozaki

  committed
  124f9bf

  Copy full SHA for 124f9bf

• Removed jupyter notebooks

  isamu-isozaki

  committed
  9900d99

  Copy full SHA for 9900d99

• Fixed merge bugs

  isamu-isozaki

  committed
  604f0dc

  Copy full SHA for 604f0dc

Commits on Mar 28, 2023

• Fixed layer norm

  isamu-isozaki

  committed
  e8ceeb2

  Copy full SHA for e8ceeb2

Commits on Apr 6, 2023

• Fixed merge conflicts

  isamu-isozaki

  committed
  d78c68e

  Copy full SHA for d78c68e

• Removed inference script

  isamu-isozaki

  committed
  baaeca0

  Copy full SHA for baaeca0