patch for gemma cpt

2024-03-12 21:21:54 +08:00
parent 60cc17f3a8
commit 70a3052dd8
2 changed files with 5 additions and 1 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -503,7 +503,7 @@ register_model_group(
        },
        "OLMo-7B-Chat": {
            DownloadSource.DEFAULT: "allenai/OLMo-7B-Instruct",
-        }
+        },
    },
    module="att_proj",
    template="olmo",