transformers
0c4a2024 - [tests] Add Context-parallel CI tests (#41860)

Commit
39 days ago
[tests] Add Context-parallel CI tests (#41860) * intial * simplify tests * add test_cp_equivalence * removed fsdp_transformer_layer_cls_to_wrap * use DataCollatorForLanguageModeling * remove use_cache=False. * changes from review * make script self contained * moved to fsdp folder * fix class name
Author
Parents
Loading