From ac1520cb8f453943d0dff9cad6a708116af3c0a6 Mon Sep 17 00:00:00 2001 From: YeAnbang Date: Mon, 3 Jun 2024 05:45:00 +0000 Subject: [PATCH] remove baichuan from template test due to transformer version conflict --- applications/ColossalChat/tests/test_templating.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/applications/ColossalChat/tests/test_templating.sh b/applications/ColossalChat/tests/test_templating.sh index 1ae93bd7e..bdb447dcf 100755 --- a/applications/ColossalChat/tests/test_templating.sh +++ b/applications/ColossalChat/tests/test_templating.sh @@ -6,8 +6,8 @@ TEST_DATA_DIR=$BASE_DIR/tests/test_data DATA_SAVE_PATH=$BASE_TEMP_DIR/tests CONFIG_DIR=$BASE_DIR/config -# MODELS=("colossal-llama2" "llama2" "mistral" "chatGLM2" "chatGLM3" "deepseek" "Yi") -MODELS=("colossal-llama2" "llama2" "chatGLM2" "chatGLM3" "deepseek" "Yi" "baichuan") +# MODELS=("colossal-llama2" "llama2" "mistral" "chatGLM2" "chatGLM3" "deepseek" "Yi" "baichuan") +MODELS=("colossal-llama2" "llama2" "chatGLM2" "chatGLM3" "deepseek" "Yi") get_pretrain() { local model=$1