17 Commits

Author	SHA1	Message	Date
Vivek Khandelwal	11f62d7fac	Minor fixes for MiniLM Training	2023-07-18 17:16:44 +05:30
Vivek Khandelwal	495670d9b6	Fix SD fine tuning script device arg usage	2023-03-10 18:37:53 +05:30
Vivek Khandelwal	cc0ef54e0e	Fix Stable diffusion fine tuning script	2023-03-06 17:52:16 +05:30
Daniel Garvey	812152485d	temporarily xfail tiny convnext macos (#1142)	2023-03-03 13:30:56 -06:00
Vivek Khandelwal	0816fb403a	Add Stable diffusion fine tuning script ... This commit adds the sd fine tuning script which runs through the torchdynamo path.	2023-03-03 21:59:00 +05:30
Ean Garvey	a504903dd5	Fix formatting issues. (#903)	2023-02-01 09:12:45 +05:30
stanley	c4a9365aa1	[Shark][Training] Refresh SharkTrainer to latest APIs.	2023-01-19 20:30:15 +00:00
Ean Garvey	45677c1e23	Install torch version required by torch-mlir when setting up importer venv. (#486)	2022-11-14 14:01:01 +05:30
mariecwhite	ec461a4456	Enable XLA compiler for TF models (#484)	2022-11-13 20:10:47 -08:00
Vivek Khandelwal	0e217d6180	Add Stable Diffusion Img2Img model script	2022-10-13 21:56:46 +05:30
Prashant Kumar	0dcf387089	Add shark_importer for torch_models. (#183) ... All the torch_models are imported to gs::shark_tank. Scripts have been updated.	2022-07-12 20:38:19 -07:00
Anush Elangovan	a7435973d9	Fix black formatting	2022-06-30 20:42:02 +00:00
Prashant Kumar	e8aa105b2a	Divide iree_utils and do module imports on function calls.	2022-06-22 14:17:33 +05:30
Prashant Kumar	4d3acff484	Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)	2022-06-16 07:28:00 -07:00
Stanley Winata	c70ab55717	Add saving and loading of mlir to run. (#89)	2022-06-05 08:42:40 -07:00
Ean Garvey	8ad73d365c	Add support for training TF models + fix TF BERT training example (#80)	2022-06-01 18:46:07 -07:00
Prashant Kumar	c69baa3b1e	Rename and refactoring.	2022-05-27 15:17:08 +05:30