diff --git a/.github/scripts/oc_score_assert.py b/.github/scripts/oc_score_assert.py index 1397db88..f869b157 100644 --- a/.github/scripts/oc_score_assert.py +++ b/.github/scripts/oc_score_assert.py @@ -16,10 +16,10 @@ chat_model_list = [ 'llama-3-8b-instruct-turbomind', 'mistral-7b-instruct-v0.2-hf', 'mistral-7b-instruct-v0.2-vllm', 'minicpm-2b-dpo-fp32-hf', 'minicpm-2b-sft-bf16-hf', 'minicpm-2b-sft-fp32-hf', - 'phi-3-mini-4k-instruct-hf', 'phi-3-small-8k-instruct-hf', - 'qwen1.5-0.5b-chat-hf', 'qwen2-1.5b-instruct-turbomind', - 'qwen2-7b-instruct-turbomind', 'qwen1.5-0.5b-chat-vllm', - 'yi-1.5-6b-chat-hf', 'yi-1.5-9b-chat-hf', 'lmdeploy-api-test' + 'phi-3-mini-4k-instruct-hf', 'qwen1.5-0.5b-chat-hf', + 'qwen2-1.5b-instruct-turbomind', 'qwen2-7b-instruct-turbomind', + 'qwen1.5-0.5b-chat-vllm', 'yi-1.5-6b-chat-hf', 'yi-1.5-9b-chat-hf', + 'lmdeploy-api-test' ] base_model_list = [ 'deepseek-moe-16b-base-hf', 'deepseek-7b-base-turbomind', diff --git a/.github/workflows/daily-run-test.yml b/.github/workflows/daily-run-test.yml index 189d6e04..7d7affaf 100644 --- a/.github/workflows/daily-run-test.yml +++ b/.github/workflows/daily-run-test.yml @@ -68,6 +68,7 @@ jobs: FLASH_ATTENTION_FORCE_BUILD=TRUE pip install /cpfs01/user/qa-llm-cicd/packages/flash_attn-2.6.3+cu118torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl pip install /cpfs01/user/qa-llm-cicd/packages/xformers-0.0.27.post2+cu118-cp310-cp310-manylinux2014_x86_64.whl --cache-dir ${{env.PIP_CACHE_PATH}} conda info --envs + pip list - name: Prepare - prepare data and hf model run: | ln -s ${{env.DATEASET_CACHE_PATH}} data diff --git a/opencompass/models/lmdeploy_pytorch.py b/opencompass/models/lmdeploy_pytorch.py index 92118066..bb195b64 100644 --- a/opencompass/models/lmdeploy_pytorch.py +++ b/opencompass/models/lmdeploy_pytorch.py @@ -88,7 +88,7 @@ class LmdeployPytorchModel(BaseModel): gen_config.stop_words = stop_words self.gen_config = gen_config self.end_str = end_str - self.major_version, self.minor_version, _ = version_info + self.major_version, self.minor_version = version_info[:2] def generate( self,