Merge branch 'main' into update-deps

.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:

.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:

.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:

CODEOWNERS

@@ -8,4 +8,4 @@
 #  https://help.github.com/en/articles/about-code-owners
 #
 
-*  @anhuong @Ssukriti @alex-jw-brooks
+*  @anhuong @Ssukriti @aluu317 @fabianlim @kmehant

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:]