Megatron-LM
Megatron-LM copied to clipboard
[BUG] context manager syntax bug in transformer_block.py
please see https://github.com/NVIDIA/Megatron-LM/pull/902