add falcon 11b

Former-commit-id: d77bed4091
2025-12-14 10:56:56 +08:00 · 2024-05-17 00:08:33 +08:00
parent 5eb8107db2
commit 22f71c152a
3 changed files with 5 additions and 2 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -375,6 +375,9 @@ register_model_group(
            DownloadSource.DEFAULT: "tiiuae/falcon-7b",
            DownloadSource.MODELSCOPE: "AI-ModelScope/falcon-7b",
        },
+        "Falcon-11B": {
+            DownloadSource.DEFAULT: "tiiuae/falcon-11B",
+        },
        "Falcon-40B": {
            DownloadSource.DEFAULT: "tiiuae/falcon-40b",
            DownloadSource.MODELSCOPE: "AI-ModelScope/falcon-40b",