Repository navigation

Flax pipeline pndm - #583

Merged

patrickvonplaten merged 30 commits into

mainfrom

flax_pipeline_pndm

Sep 27, 2022

Merged

Flax pipeline pndm#583
patrickvonplaten merged 30 commits into
mainfrom
flax_pipeline_pndm

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.
  pcuenca
  committed
  4dfcf21
- Convert params to bfloat16 or fp16 after loading.
  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.
  pcuenca
  committed
  aa3c010
- Pass latents shape to scheduler set_timesteps()
  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.
  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.
  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
  
  pcuenca
  and
  patil-suraj
  authored
  f974a41
- Apply suggestion list -> tuple
  
  pcuenca
  and
  patil-suraj
  authored
  ce0a327
- Remove unused comments.
  pcuenca
  committed
  7fcbc32
- Use zeros instead of empty.
  pcuenca
  committed
  cd17c56

Commits

Commits on Sep 19, 2022

Commits on Sep 20, 2022

Commits on Sep 21, 2022

Commits on Sep 22, 2022