summaryrefslogtreecommitdiffstats
path: root/training/util.py
Commit message (Expand)AuthorAgeFilesLines
* UpdateVolpeon2023-01-061-6/+6
* Use context manager for EMA, on_train/eval hooksVolpeon2023-01-061-1/+1
* Add contextmanager to EMAModel to apply weights temporarilyVolpeon2023-01-061-0/+12
* Added EMA to TIVolpeon2023-01-051-5/+95
* Added multi-vector embeddingsVolpeon2022-12-311-4/+56
* Integrated updates from diffusersVolpeon2022-12-281-5/+0
* Code simplifications, avoid autocastVolpeon2022-12-251-57/+49
* FixVolpeon2022-12-241-1/+1
* Fix againVolpeon2022-12-241-1/+1
* FixVolpeon2022-12-241-1/+4
* Improved class prompt handlingVolpeon2022-12-231-3/+1
* Moved common training code into separate moduleVolpeon2022-12-211-0/+131