diff --git a/codecheck_toolkits/mypy.sh b/codecheck_toolkits/mypy.sh index b91efeff7130a1800473bdfc8b77410beff5afad..9706d7e00c080ca66e9b9da61db4a6e0177ecbfa 100755 --- a/codecheck_toolkits/mypy.sh +++ b/codecheck_toolkits/mypy.sh @@ -9,7 +9,7 @@ run_mypy() { } run_mypy -run_mypy tests +run_mypy tests/st run_mypy vllm_mindspore/attention run_mypy vllm_mindspore/compilation run_mypy vllm_mindspore/distributed diff --git a/requirements.txt b/requirements.txt index 13ff48419ec0d0c5df921bbad689f1544b4947df..fe476f1b7eb07509c8bab3694166f5f79c11bc02 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,3 @@ -ray \ No newline at end of file +ray +# transformers 4.52.x add torch checker, limit transformers version +transformers==4.51.3