From 32b05d07fd2446200e09f15de92ef3f456e2828c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=95=96=E6=AD=A6?= Date: Thu, 8 Jun 2023 16:01:51 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=96=B0=20demo=EF=BC=8C?= =?UTF-8?q?=E6=94=AF=E6=8C=81=E8=BF=90=E8=A1=8C=20moss-moon-003-sft-plugin?= =?UTF-8?q?=20=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- moss_api_demo.py | 5 +++-- moss_cli_demo.py | 3 ++- moss_cli_demo_jittor.py | 3 ++- moss_web_demo_gradio.py | 5 +++-- moss_web_demo_streamlit.py | 5 +++-- 5 files changed, 13 insertions(+), 8 deletions(-) diff --git a/moss_api_demo.py b/moss_api_demo.py index 3c1bad2..dee8fe5 100644 --- a/moss_api_demo.py +++ b/moss_api_demo.py @@ -21,8 +21,9 @@ parser = argparse.ArgumentParser() parser.add_argument("--model_name", default="fnlp/moss-moon-003-sft-int4", - choices=["fnlp/moss-moon-003-sft", - "fnlp/moss-moon-003-sft-int8", + choices=["fnlp/moss-moon-003-sft", + "fnlp/moss-moon-003-sft-plugin", + "fnlp/moss-moon-003-sft-int8", "fnlp/moss-moon-003-sft-int4"], type=str) parser.add_argument("--gpu", default="0", type=str) args = parser.parse_args() diff --git a/moss_cli_demo.py b/moss_cli_demo.py index bb36e40..eec44f8 100644 --- a/moss_cli_demo.py +++ b/moss_cli_demo.py @@ -14,7 +14,8 @@ parser = argparse.ArgumentParser() parser.add_argument("--model_name", default="fnlp/moss-moon-003-sft-int4", - choices=["fnlp/moss-moon-003-sft", + choices=["fnlp/moss-moon-003-sft", + "fnlp/moss-moon-003-sft-plugin", "fnlp/moss-moon-003-sft-int8", "fnlp/moss-moon-003-sft-int4"], type=str) parser.add_argument("--gpu", default="0", type=str) diff --git a/moss_cli_demo_jittor.py b/moss_cli_demo_jittor.py index b684b7c..0ede867 100644 --- a/moss_cli_demo_jittor.py +++ b/moss_cli_demo_jittor.py @@ -14,7 +14,8 @@ parser = argparse.ArgumentParser() parser.add_argument("--model_name", default="fnlp/moss-moon-003-sft", - choices=["fnlp/moss-moon-003-sft", + choices=["fnlp/moss-moon-003-sft", + "fnlp/moss-moon-003-sft-plugin", "fnlp/moss-moon-003-sft-int8", "fnlp/moss-moon-003-sft-int4"], type=str) parser.add_argument("--generate", default="sample", diff --git a/moss_web_demo_gradio.py b/moss_web_demo_gradio.py index 5e819da..649424e 100644 --- a/moss_web_demo_gradio.py +++ b/moss_web_demo_gradio.py @@ -20,8 +20,9 @@ parser = argparse.ArgumentParser() parser.add_argument("--model_name", default="fnlp/moss-moon-003-sft-int4", - choices=["fnlp/moss-moon-003-sft", - "fnlp/moss-moon-003-sft-int8", + choices=["fnlp/moss-moon-003-sft", + "fnlp/moss-moon-003-sft-plugin", + "fnlp/moss-moon-003-sft-int8", "fnlp/moss-moon-003-sft-int4"], type=str) parser.add_argument("--gpu", default="0", type=str) args = parser.parse_args() diff --git a/moss_web_demo_streamlit.py b/moss_web_demo_streamlit.py index 279cb05..59aae87 100644 --- a/moss_web_demo_streamlit.py +++ b/moss_web_demo_streamlit.py @@ -15,8 +15,9 @@ parser = argparse.ArgumentParser() parser.add_argument("--model_name", default="fnlp/moss-moon-003-sft-int4", - choices=["fnlp/moss-moon-003-sft", - "fnlp/moss-moon-003-sft-int8", + choices=["fnlp/moss-moon-003-sft", + "fnlp/moss-moon-003-sft-plugin", + "fnlp/moss-moon-003-sft-int8", "fnlp/moss-moon-003-sft-int4"], type=str) parser.add_argument("--gpu", default="0", type=str) args = parser.parse_args()