7 Commits

Author	SHA1	Message	Date
Shivam Mehta	39cbd85236	Using Wget for new ckpt downloadsA	2024-01-12 11:09:25 +00:00
mush42	336dd20d5b	Use torch.onnx.is_in_onnx_export() instead of torch.jit.is_scripting() since the former is dedicated to this use case.	2023-09-26 15:28:15 +02:00
mush42	01c99161c4	- Fixed several bugs. Thanks @shivammehta25 for the suggestions	2023-09-26 14:21:17 +02: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	acb57a3857	Minor release till I release checkpoints and urls	2023-09-16 20:37:11 +00:00
Shivam Mehta	f016784049	Initial commit	2023-09-16 17:51:36 +00:00