ColossalAI/tests/test_shardformer/test_layer
flybird11111 79718fae04
[shardformer] llama support DistCrossEntropy (#5176)
* fix

aaa

fix

fix

fix

* fix

* fix

* test ci

* fix ci

fix

* llama support dist-cross

fix

fix

fix

fix

fix

fix

fix

fix

* fix

* fix

* fix

fix

* test ci

* test ci

* fix

* [Colossal-Llama-2] Add finetuning Colossal-Llama-2 example (#4878)

* Add finetuning Colossal-Llama-2 example

* Add finetuning Colossal-Llama-2 example 2

* Add finetuning Colossal-Llama-2 example and support NEFTuning

* Add inference example and refine neftune

* Modify readme file

* update the imports

---------

Co-authored-by: Xu Yuanchen <yuanchen.xu00@gmail.com>
Co-authored-by: Camille Zhong <44392324+Camille7777@users.noreply.github.com>

* llama support dist-cross

fix

fix

fix

fix

fix

fix

fix

fix

* fix

* fix

* fix

fix

* test ci

* test ci

* fix

* fix ci

* fix ci

---------

Co-authored-by: Yuanchen <70520919+chengeharrison@users.noreply.github.com>
Co-authored-by: Xu Yuanchen <yuanchen.xu00@gmail.com>
Co-authored-by: Camille Zhong <44392324+Camille7777@users.noreply.github.com>
2023-12-13 01:39:14 +08:00
..
test_dist_crossentropy.py [shardformer] llama support DistCrossEntropy (#5176) 2023-12-13 01:39:14 +08:00
test_dropout.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_embedding.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_gpt2_qkv_fused_linear_1d.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_layernorm.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_linear_1d.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_qkv_fused_linear_1d.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_vocab_parallel_embedding_1d.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00