Skip to content

GitLab

Commit 48b2dafd authored by dongcl's avatar dongcl
Browse files

support 1f1b overlap

parent c63fceee
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
dcu_megatron/adaptor/megatron_adaptor.py
View file @ 48b2dafd
......@@ -25,11 +25,11 @@ class MegatronAdaptation:
MegatronAdaptation.apply()
# apply features
# from .patch_utils import MegatronPatchesManager
# from .features_manager import a2a_overlap_adaptation
from .patch_utils import MegatronPatchesManager
from .features_manager import a2a_overlap_adaptation
# a2a_overlap_adaptation(MegatronPatchesManager)
# MegatronPatchesManager.apply_patches()
a2a_overlap_adaptation(MegatronPatchesManager)
MegatronPatchesManager.apply_patches()
@classmethod
def register(cls, orig_func_name, new_func=None, force_patch=False, create_dummy=False, apply_wrapper=False, remove_origin_wrappers=False):
......
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