3 Commits

Author	SHA1	Message	Date
Pedro Cuenca	92e1164e2e	`mps`: remove warmup passes (#2771) ... * Remove warmup passes in mps tests. * Update mps docs: no warmup pass in PyTorch 2 * Update imports.	2023-03-22 19:29:27 +01:00
Patrick von Platen	eadf0e2555	[Copyright] 2023 (#2524)	2023-03-01 10:31:00 +01:00
Patrick von Platen	a80480f0f2	[Tests] Improve unet / vae tests (#1018) ... * improve tests * up * finish * upload * add init * up * finish vae * finish * reduce loading time with device_map * remove device_map from CPU * uP	2022-10-28 13:43:26 +02:00