summaryrefslogtreecommitdiffstats
path: root/training/util.py
Commit message (Collapse)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