transformers/tests/models/megatron_gpt2
..
__init__.py
test_modeling_megatron_gpt2.py