Keyan (Kyrie) Zhang 2e80701f58 Unit test on final logits and the logits of the last attention layer (#10093) ... * Add unit test on final logits and attention * Add unit test on final logits and attention * Modify unit test on final logits and attention		2024-02-07 14:25:36 +08:00
..
test_layer_fast_rope.py	Support fast rope for training (#9745)	2024-01-17 15:51:38 +08:00
test_transformers_api.py	Unit test on final logits and the logits of the last attention layer (#10093)	2024-02-07 14:25:36 +08:00
test_transformers_api_attention.py	Unit test on final logits and the logits of the last attention layer (#10093)	2024-02-07 14:25:36 +08:00
test_transformers_api_final_logits.py	Unit test on final logits and the logits of the last attention layer (#10093)	2024-02-07 14:25:36 +08:00