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()