From a37f07472fd4fd3272769a21d1bba718d8b2e5ca Mon Sep 17 00:00:00 2001 From: Will Date: Fri, 27 Sep 2024 19:02:24 -0400 Subject: [PATCH 1/3] fix: check for wte.weight along with embed_tokens.weight (#356) Signed-off-by: Will Johnson --- tuning/utils/merge_model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tuning/utils/merge_model_utils.py b/tuning/utils/merge_model_utils.py index a2ec6577..c2078908 100644 --- a/tuning/utils/merge_model_utils.py +++ b/tuning/utils/merge_model_utils.py @@ -168,7 +168,7 @@ def post_process_vLLM_adapters_new_tokens( # vLLM requires renaming to output_embeddings new_embeddings["output_embeddings"] = new_output_embeddings - elif "embed_tokens.weight" in k: + elif "embed_tokens.weight" in k or "wte.weight" in k: embed_tokens = f.get_tensor(k) # pull out tensor values of new tokens new_input_embeddings = embed_tokens[-num_added_tokens:] From 9243be75ddde7a2ba6a9add8e3b968538f8fef6b Mon Sep 17 00:00:00 2001 From: Anh Uong Date: Tue, 1 Oct 2024 09:58:32 -0600 Subject: [PATCH 2/3] ci: run unit tests, fmt, image build on release branch (#361) Signed-off-by: Anh Uong --- .github/workflows/format.yml | 4 ++-- .github/workflows/image.yaml | 4 ++-- .github/workflows/test.yaml | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/workflows/format.yml b/.github/workflows/format.yml index 574aac18..6e26a490 100644 --- a/.github/workflows/format.yml +++ b/.github/workflows/format.yml @@ -16,9 +16,9 @@ name: Format on: push: - branches: [ "main" ] + branches: [ "main", "release" ] pull_request: - branches: [ "main" ] + branches: [ "main", "release" ] jobs: lint: diff --git a/.github/workflows/image.yaml b/.github/workflows/image.yaml index d4d836be..6c1e043c 100644 --- a/.github/workflows/image.yaml +++ b/.github/workflows/image.yaml @@ -1,9 +1,9 @@ name: Image on: push: - branches: [ "main" ] + branches: [ "main", "release" ] pull_request: - branches: [ "main" ] + branches: [ "main", "release" ] jobs: build: diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index f8e24265..ead02bb3 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -1,9 +1,9 @@ name: Test on: push: - branches: [ "main" ] + branches: [ "main", "release" ] pull_request: - branches: [ "main" ] + branches: [ "main", "release" ] jobs: build: From 7b97e9e478385b21b597ebb9c72ae75426e5ec11 Mon Sep 17 00:00:00 2001 From: Anh Uong Date: Wed, 2 Oct 2024 10:07:36 -0600 Subject: [PATCH 3/3] chore: update code owners (#363) Signed-off-by: Anh Uong --- CODEOWNERS | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CODEOWNERS b/CODEOWNERS index a28fcff9..bcd048d9 100644 --- a/CODEOWNERS +++ b/CODEOWNERS @@ -8,4 +8,4 @@ # https://help.github.com/en/articles/about-code-owners # -* @anhuong @Ssukriti @alex-jw-brooks +* @anhuong @Ssukriti @aluu317 @fabianlim @kmehant