diff --git a/OWNERS b/OWNERS index 9cfc42f2f..3b419dcbb 100644 --- a/OWNERS +++ b/OWNERS @@ -2,6 +2,7 @@ approvers: - fengliangjun66 - wujianping996 - dilililiwhy +- liuyanghan reviewers: - fengliangjun66 - guhangsong diff --git a/tests/ut/test_convert_ckpt_from_huggingface.py b/tests/ut/test_convert_ckpt_from_huggingface.py index eee614cd6..4747893a9 100644 --- a/tests/ut/test_convert_ckpt_from_huggingface.py +++ b/tests/ut/test_convert_ckpt_from_huggingface.py @@ -51,9 +51,9 @@ class CovertMCoreCkptFromHuggingfaceArgs: saver = "mg_mcore" target_tensor_parallel_size = "2" target_pipeline_parallel_size = "4" - load_dir = "/home/dataset/ci_engineering/llama-2-7b-hf/" - save_dir = "/home/dataset/ci_engineering/llama-2-7b-mg-tp2-pp4-mcore-test/" - base_dir = "/home/dataset/ci_engineering/llama-2-7b-mg-tp2-pp4-mcore/" + load_dir = "/data/llama-2-7b-hf/" + save_dir = "/data/llama-2-7b-mg-tp2-pp4-mcore-test/" + base_dir = "/data/llama-2-7b-mg-tp2-pp4-mcore/" tokenizer_model = "/home/dataset/ci_engineering/llama-2-7b-hf/tokenizer.model" diff --git a/tools/checkpoint/loader_hf_mcore.py b/tools/checkpoint/loader_hf_mcore.py index 5d0dbf4f0..46ceeade1 100644 --- a/tools/checkpoint/loader_hf_mcore.py +++ b/tools/checkpoint/loader_hf_mcore.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +# Copyright (c) 2023, NVIDIA CORPORATION. All rights reserved. import os import sys