ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add benchmark_util for transformers >= 4.47.0 (#12644)

Browse source
This commit is contained in:
Zijie Li 2025-01-03 10:48:29 +08:00 committed by GitHub
parent 550fa01649
commit 8fd2dcba86
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 4910 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/dev/benchmark/all-in-one/run.py
View file

@ -32,10 +32,7 @@ current_dir = os.path.dirname(os.path.realpath(__file__))
import sys
sys.stdout.reconfigure(encoding='utf-8')
try:
from ipex_llm.utils import BenchmarkWrapper
except:
pass
from ipex_llm.utils.common.log4Error import invalidInputError
from ipex_llm.utils.common import invalidInputError

2
python/llm/dev/test/lint-python
View file

@ -21,7 +21,7 @@ SCRIPT_DIR="$( cd "$( dirname "$0" )" && pwd )"
PYTHON_ROOT_DIR="$SCRIPT_DIR/.."
echo $PYTHON_ROOT_DIR
PATHS_TO_CHECK="$SCRIPT_DIR/../../src"
PATTERNS_TO_EXCLUDE="__init__.py,log4Error.py,$SCRIPT_DIR/../../src/ipex_llm/langchain/*,$SCRIPT_DIR/../../src/ipex_llm/transformers/gguf/models/model_implement/yuan2/*,benchmark_util_4_29.py,benchmark_util_4_42.py,benchmark_util_4_43.py,benchmark_util_4_44.py,benchmark_util_4_45.py,tgi_api_server.py,api_server.py"
PATTERNS_TO_EXCLUDE="__init__.py,log4Error.py,$SCRIPT_DIR/../../src/ipex_llm/langchain/*,$SCRIPT_DIR/../../src/ipex_llm/transformers/gguf/models/model_implement/yuan2/*,benchmark_util_4_29.py,benchmark_util_4_42.py,benchmark_util_4_43.py,benchmark_util_4_44.py,benchmark_util_4_45.py,benchmark_util_4_47.py,tgi_api_server.py,api_server.py"
PEP8_REPORT_PATH="$PYTHON_ROOT_DIR/test/pep8-report.txt"
PYLINT_REPORT_PATH="$PYTHON_ROOT_DIR/test/pylint-report.txt"
PYLINT_INSTALL_INFO="$PYTHON_ROOT_DIR/test/pylint-info.txt"

3
python/llm/src/ipex_llm/utils/__init__.py
View file

@ -23,8 +23,7 @@ import transformers
trans_version = transformers.__version__
if trans_version >= "4.47.0":
# TODO
pass
from .benchmark_util_4_47 import BenchmarkWrapper
elif trans_version >= "4.45.0":
from .benchmark_util_4_45 import BenchmarkWrapper
elif trans_version >= "4.44.0":

4907
python/llm/src/ipex_llm/utils/benchmark_util_4_47.py Normal file
View file

File diff suppressed because it is too large Load diff