10 Commits

Author	SHA1	Message	Date
Shivam Mehta	aa496aa13f	Adding the possibility to train with durations	2024-05-27 13:24:21 +02:00
Shivam Mehta	4b39f6cad0	Adding the possibility of get durations out of pretrained model	2024-05-24 11:34:51 +02:00
Shivam Mehta	0ed9290c31	Logging global step while training	2023-12-06 10:39:54 +00:00
Shivam Mehta	f39ee6cf3b	Changing while to for for more readibility	2023-12-05 12:10:52 +00:00
Shivam Mehta	6e71dc8b8f	adding prior loss as a configuration	2023-12-05 09:57:37 +00:00
mush42	1b204ed42c	ONNX export and inference. Complete and tested implmentation.	2023-09-24 01:57:35 +02:00
Shivam Mehta	a9251ed984	Adding docstrings	2023-09-17 06:50:46 +00:00
Shivam Mehta	c079e5254a	Adding docstrings	2023-09-17 06:49:12 +00:00
Shivam Mehta	88bc7d05eb	Moving from diffusers to incode	2023-09-16 19:57:12 +00:00
Shivam Mehta	f016784049	Initial commit	2023-09-16 17:51:36 +00:00