This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#checking if conversion is correct | |
code: | |
import sys | |
import torch | |
from megatron import get_args, get_tokenizer, initialize_megatron, mpu | |
from megatron.model import GPTModel | |
from megatron.training import get_model | |
from megatron.checkpointing import load_checkpoint |
NewerOlder