Commit 941a793f authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

removed unused 'get_args' import

parent 5993f04b

megatron/mpu/random.py

+0 −1
Original line number Diff line number Diff line
@@ -24,7 +24,6 @@ from torch import _C 
from torch.cuda import _lazy_call, device as device_ctx_manager

from torch.utils.checkpoint import detach_variable



from megatron import get_args

from megatron.memory import allocate_mem_buff



from .initialize import get_data_parallel_rank