Skip to content

GitLab

Commit 77aad5e5 authored by dongcl's avatar dongcl
Browse files

bug fix

parent 7c941853
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
dcu_megatron/legacy/model/transformer.py
View file @ 77aad5e5
......@@ -4,6 +4,7 @@ import torch.nn.functional as F
from megatron.training import get_args
from megatron.core import tensor_parallel
from megatron.legacy.model.module import MegatronModule
from megatron.legacy.model.transformer import ParallelMLP
from megatron.legacy.model.utils import (
erf_gelu,
openai_gelu,
......
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