Edenzzzz dcc44aab8d [misc] Use dist logger in plugins (#6011) ...

* use dist logger in plugins

* remove trash

* print on rank 0

---------

Co-authored-by: Edenzzzz <wtan45@wisc.edu>

2024-08-20 10:32:41 +08:00

..

gemini

[misc] Use dist logger in plugins (#6011)

2024-08-20 10:32:41 +08:00

low_level

[test] fix test: test_zero1_2

2024-08-01 10:06:59 +08:00

__init__.py

[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)

2023-11-28 16:54:42 +08:00

wrapper.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00