Code

Flax pipeline pndm#583

Code

Merged

patrickvonplaten merged 30 commits intomainhuggingface/diffusers:mainfrom flax_pipeline_pndmhuggingface/diffusers:flax_pipeline_pndmCopy head branch name to clipboardSep 27, 2022

+137-54

Commits

Commits on Sep 19, 2022

• WIP: flax FlaxDiffusionPipeline & FlaxStableDiffusionPipeline

  mishig25

  committed
  b9ca406

• todo comment

  mishig25

  committed
  30abc63

• Merge branch 'main' into flax_pipeline

  

  Mishig Davaadorj

  authored
  9b54559

• Fix imports

  mishig25

  committed
  4b2becb

• Fix imports

  mishig25

  committed
  7f0e429

• add dummies

  patrickvonplaten

  committed
  d9e2ae1

• Fix empty init

  mishig25

  committed
  d51e881

• Merge branch 'flax_pipeline' of https://github.com/huggingface/diffusers into flax_pipeline

  mishig25

  committed
  741046d

• make pipeline work

  patrickvonplaten

  committed
  7aab68d

• merge conflict

  patrickvonplaten

  committed
  7d3fff6

• up

  patrickvonplaten

  committed
  47d7739

Commits on Sep 20, 2022

• Allow dtype to be overridden on model load.

  Show description for 4dfcf21
  pcuenca

  committed
  4dfcf21

• Convert params to bfloat16 or fp16 after loading.

  Show description for d480534
  pcuenca

  committed
  d480534

• Use Flax schedulers (typing, docstring)

  pcuenca

  committed
  0c2a868

• Merge branch 'flax_pipeline' into flax_pipeline_bf16

  pcuenca

  committed
  a71e6be

• PNDM: replace control flow with jax functions.

  Show description for aa3c010
  pcuenca

  committed
  aa3c010

• Pass latents shape to scheduler set_timesteps()

  Show description for d6dbb89
  pcuenca

  committed
  d6dbb89

• Wrap model imports inside availability checks.

  pcuenca

  committed
  69b1d7a

• Merge branch 'flax_pipeline' into flax_pipeline_pndm

  pcuenca

  committed
  7091c1d

• Optionally return state in from_config.

  Show description for 23f7d73
  pcuenca

  committed
  23f7d73

• Do not convert model weights to dtype.

  pcuenca

  committed
  163df38

• Merge branch 'flax_pipeline_bf16' into flax_pipeline_pndm

  pcuenca

  committed
  039d1d6

• Re-enable PRK steps with functional implementation.

  Show description for 8bc06b0
  pcuenca

  committed
  8bc06b0

Commits on Sep 21, 2022

• Merge remote-tracking branch 'origin/main' into flax_pipeline_pndm

  pcuenca

  committed
  3752bbc

• Remove left over has_state var.

  pcuenca

  committed
  8a9ccf2

• make style

  pcuenca

  committed
  cf6cd7a

Commits on Sep 22, 2022

• Apply suggestion list -> tuple

  Show description for f974a41
  

  pcuenca

  and

  patil-suraj

  authored
  f974a41

• Apply suggestion list -> tuple

  Show description for ce0a327
  

  pcuenca

  and

  patil-suraj

  authored
  ce0a327

• Remove unused comments.

  pcuenca

  committed
  7fcbc32

• Use zeros instead of empty.

  pcuenca

  committed
  cd17c56