GitLab

Temporarily remove Flax modules from the public API

* up

* uP

* uP

* make style

* Apply suggestions from code review

* up

* finish

* [CI] Localize the HF cache

* pip cache

* de-env

* refactor matrix

* fix fast cache

* less onnx steps

* revert

* revert pip cache

* revert pip cache

* remove debugging trigger

* Fix the LMS pytorch regression

* Copy over the changes from #637

* Copy over the changes from #637

* Fix betas test

* Handle the PIL.Image.Resampling deprecation

* style

* start

* finish

* Update black

* update table

uP

* Init CI

* clarify cpu

* style

* Check scripts quality too

* Drop smi for cpu tests

* Run PR tests on cpu docker envs

* Update .github/workflows/push_tests.yml
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* Try minimal python container

* Print env, install stable GPU torch

* Manual torch install

* remove deprecated platform.dist()
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

Pin black==22.3

* Add is_<framework>_available, refactor import utils

* deps

* quality

* Revive Make utils

* Add datasets for training too

* add a metrics logger

* fix LatentDiffusionUncondPipeline

* add VQModel in init

* add image logging to tensorboard

* switch manual templates to the modelcards package

* hide ldm example
Co-authored-by: patil-suraj <surajp815@gmail.com>