Yih-Dar 6ea3ee3cd2 Fix test_model_parallelism (#25359) ...

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

2023-08-08 10:48:45 +02:00

..

__init__.py

Add OPT (#17088)

2022-05-12 12:24:35 +02:00

test_modeling_flax_opt.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00

test_modeling_opt.py

Fix test_model_parallelism (#25359)

2023-08-08 10:48:45 +02:00

test_modeling_tf_opt.py

Big TF test cleanup (#24282)

2023-06-16 15:40:49 +01:00