Frank Lee c4b1b65931 [test] fixed tests failed due to dtensor change (#4082) ... * [test] fixed tests failed due to dtensor change * polish code		2023-07-04 16:05:01 +08:00
..
benchmark_autochunk_diffuser.py	[autochunk] support complete benchmark (#3121)	2023-03-13 17:42:37 +08:00
test_autochunk_diffuser_utils.py	[devops] update torch version of CI (#3725)	2023-05-15 17:20:56 +08:00
test_autochunk_unet.py	[test] fixed tests failed due to dtensor change (#4082)	2023-07-04 16:05:01 +08:00