From 3c3bc98f5ef0da76491cf69029d41f73d96ed531 Mon Sep 17 00:00:00 2001 From: wql Date: Tue, 22 Oct 2024 17:25:10 +0800 Subject: [PATCH] feat: add constants for 9g --- src/llamafactory/extras/constants.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py index 80df135c..2da007ae 100644 --- a/src/llamafactory/extras/constants.py +++ b/src/llamafactory/extras/constants.py @@ -804,6 +804,15 @@ register_model_group( template="cpm", ) +register_model_group( + models={ + "CPM-9G-8B": { + DownloadSource.DEFAULT: "NotImplemented/UseFilePath", + DownloadSource.MODELSCOPE: "NotImplemented/UseFilePath", + }, + }, + template="cpm", +) register_model_group( models={