Skip to content

GitLab

Commit 1cc21017 authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

missing import.

parent c5be45ee
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
megatron/optimizer/distrib_optimizer.py
View file @ 1cc21017
...@@ -22,6 +22,7 @@ import torch ...@@ -22,6 +22,7 @@ import torch
from megatron import get_args from megatron import get_args
from megatron import get_timers from megatron import get_timers
from megatron import mpu from megatron import mpu
from megatron import print_rank_0
from megatron.model.module import param_is_not_shared from megatron.model.module import param_is_not_shared
from megatron.mpu.layers import param_is_not_tensor_parallel_duplicate from megatron.mpu.layers import param_is_not_tensor_parallel_duplicate
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment