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	…
test_autochunk_diffuser_utils.py	…
test_autochunk_unet.py	[test] fixed tests failed due to dtensor change (#4082)	2023-07-04 16:05:01 +08:00