Skip to content
This repository was archived by the owner on Aug 30, 2024. It is now read-only.

Commit

Permalink
rename requirements of extension tests from txt to sh (#181)
Browse files Browse the repository at this point in the history
  • Loading branch information
Zhenzhong1 authored Mar 21, 2024
1 parent a85b7a5 commit 20fd168
Show file tree
Hide file tree
Showing 7 changed files with 57 additions and 8 deletions.
19 changes: 19 additions & 0 deletions neural_speed/models/requirements/baichuan13b-gptq.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
#!/bin/bash
#===============================================================================
# Copyright (c) 2023 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#===============================================================================

# To avoid the error: 'ChatGLMTokenizer' object has no attribute 'sp_tokenizer'
pip install -r "$(dirname "${BASH_SOURCE[0]}")/common.txt" transformers==4.33.1
2 changes: 0 additions & 2 deletions neural_speed/models/requirements/baichuan13b-gptq.txt

This file was deleted.

18 changes: 18 additions & 0 deletions neural_speed/models/requirements/mistral.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
#!/bin/bash
#===============================================================================
# Copyright (c) 2023 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#===============================================================================

pip install -r "$(dirname "${BASH_SOURCE[0]}")/common.txt" transformers>=4.34.0
2 changes: 0 additions & 2 deletions neural_speed/models/requirements/mistral.txt

This file was deleted.

18 changes: 18 additions & 0 deletions neural_speed/models/requirements/mixtral-gptq.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
#!/bin/bash
#===============================================================================
# Copyright (c) 2023 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#===============================================================================

pip install -r "$(dirname "${BASH_SOURCE[0]}")/common.txt" transformers>=4.34.0
2 changes: 0 additions & 2 deletions neural_speed/models/requirements/mixtral-gptq.txt

This file was deleted.

4 changes: 2 additions & 2 deletions tests/model-test/cpp_graph_inference.sh
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,7 @@ function main() {
quant_script="./build/bin/quant_mistral"
convert_script="${convert_script}/convert_mistral.py"
infer_cmd="./build/bin/run_mistral"
requirements_file="$working_dir/neural_speed/models/requirements/mistral.txt"
requirements_file="$working_dir/neural_speed/models/requirements/mistral.sh"
elif [[ "${model}" == "qwen-7b" ]]; then
quant_script="./build/bin/quant_qwen"
convert_script="${convert_script}/convert_qwen.py"
Expand Down Expand Up @@ -292,7 +292,7 @@ function main() {
elif [[ "${model}" == *"-gptq" ]]; then
infer_cmd="python $working_dir/scripts/python_api_example_for_gptq.py ${model_path}"
precision_list+=("default")
requirements_file="$working_dir/neural_speed/models/requirements/${model}.txt"
requirements_file="$working_dir/neural_speed/models/requirements/${model}.sh"
else
echo "Error: Unexpedted model: $model" 1>&2
exit 1
Expand Down

0 comments on commit 20fd168

Please sign in to comment.