This repository was archived by the owner on Oct 25, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 5 files changed +5
-5
lines changed
examples/huggingface/pytorch
code-generation/quantization
text-generation/quantization Expand file tree Collapse file tree 5 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -61,7 +61,7 @@ ENV COMPOSE_DOCKER_CLI_BUILD=0
6161# Install torch and intel-extension-for-pytorch 2.1
6262RUN python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
6363RUN python3 -m pip install intel-extension-for-pytorch intel-extension-for-transformers optimum
64- RUN python3 -m pip install git+https://github.com/huggingface/optimum-intel.git@f95dea1ae8966dee4d75d622e7b2468c514ba02d
64+ RUN python3 -m pip install git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
6565RUN python3 -m pip install git+https://github.com/bigcode-project/bigcode-evaluation-harness@0d84db85f9ff971fa23a187a3347b7f59af288dc
6666
6767# Standard requirements
Original file line number Diff line number Diff line change @@ -10,6 +10,6 @@ transformers >= 4.35.0
1010tiktoken #code_gen
1111neural-compressor
1212intel_extension_for_pytorch == 2.3.0
13- optimum-intel
13+ git+https://github.com/huggingface/ optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
1414auto-round == 0.2
1515git+https://github.com/bigcode-project/bigcode-evaluation-harness@094c7cc197d13a53c19303865e2056f1c7488ac1
Original file line number Diff line number Diff line change @@ -7,7 +7,7 @@ sentencepiece != 0.1.92
77--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
88torch==2.1.0a0
99transformers
10- optimum-intel
10+ git+https://github.com/huggingface/ optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
1111bitsandbytes #baichuan
1212transformers_stream_generator
1313tiktoken #qwen
Original file line number Diff line number Diff line change @@ -7,7 +7,7 @@ sentencepiece != 0.1.92
77torch==2.3.0+cpu
88transformers==4.38.1
99intel_extension_for_pytorch==2.3.0
10- optimum-intel
10+ git+https://github.com/huggingface/ optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
1111bitsandbytes #baichuan
1212transformers_stream_generator
1313tiktoken #qwen
Original file line number Diff line number Diff line change @@ -7,6 +7,7 @@ datasets==2.16.1
77einops
88evaluate
99gguf
10+ git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
1011git+https://github.com/intel/neural-compressor.git
1112git+https://github.com/intel/neural-speed.git
1213intel-extension-for-pytorch == 2.3.0
1617nlpaug == 1.1.9
1718onnx
1819onnxruntime
19- optimum-intel
2020peft == 0.6.2
2121py-cpuinfo
2222sacremoses
You can’t perform that action at this time.
0 commit comments