Bump release version to v3.4.1

.azure-pipelines/scripts/fwk_version.sh

@@ -2,7 +2,7 @@
 
 echo "export FWs version..."
 export tensorflow_version='2.15.0-official'
-export pytorch_version='2.5.1+cpu'
+export pytorch_version='2.5.1'
 export torchvision_version='0.20.1'
 export ipex_version='2.5.0+cpu'
 export onnx_version='1.17.0'

.azure-pipelines/scripts/ut/env_setup.sh

@@ -40,11 +40,11 @@ if [[ "${itex_version}" != "" ]]; then
 fi
 
 if [[ "${pytorch_version}" != "" ]]; then
-    pip install torch==${pytorch_version} -f https://download.pytorch.org/whl/torch_stable.html
+    pip install torch==${pytorch_version} --index-url https://download.pytorch.org/whl/cpu
 fi
 
 if [[ "${torchvision_version}" != "" ]]; then
-    pip install torchvision==${torchvision_version} -f https://download.pytorch.org/whl/torch_stable.html
+    pip install torchvision==${torchvision_version} --index-url https://download.pytorch.org/whl/cpu
 fi
 
 if [[ "${ipex_version}" != "" ]]; then

.azure-pipelines/scripts/ut/run_basic_pt_pruning.sh

@@ -4,8 +4,8 @@ test_case="run basic pt pruning"
 echo "${test_case}"
 
 echo "specify fwk version..."
-export pytorch_version='2.4.0+cpu'
-export torchvision_version='0.18.0+cpu'
+export pytorch_version='2.4.0'
+export torchvision_version='0.18.0'
 export ipex_version='2.4.0+cpu'
 
 echo "set up UT env..."

README.md

@@ -5,7 +5,7 @@ Intel® Neural Compressor
 <h3> An open-source Python library supporting popular model compression techniques on all mainstream deep learning frameworks (TensorFlow, PyTorch, and ONNX Runtime)</h3>
 
 [![python](https://img.shields.io/badge/python-3.8%2B-blue)](https://github.com/intel/neural-compressor)
-[![version](https://img.shields.io/badge/release-3.4-green)](https://github.com/intel/neural-compressor/releases)
+[![version](https://img.shields.io/badge/release-3.4.1-green)](https://github.com/intel/neural-compressor/releases)
 [![license](https://img.shields.io/badge/license-Apache%202-blue)](https://github.com/intel/neural-compressor/blob/master/LICENSE)
 [![coverage](https://img.shields.io/badge/coverage-85%25-green)](https://github.com/intel/neural-compressor)
 [![Downloads](https://static.pepy.tech/personalized-badge/neural-compressor?period=total&units=international_system&left_color=grey&right_color=green&left_text=downloads)](https://pepy.tech/project/neural-compressor)

examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/llm_quantization_recipes.md

@@ -31,7 +31,7 @@ The scripts [run_generation_cpu_woq.py](./run_generation_cpu_woq.py) provide Wei
 cd examples/huggingface/pytorch/text-generation/quantization
 pip install -r requirements.txt
 pip install neural-compressor==3.1
-pip install torch==2.4.0+cpu --index-url https://download.pytorch.org/whl/cpu
+pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
 pip install transformers==4.43.0
 ```
 

neural_compressor/version.py

@@ -15,4 +15,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Intel® Neural Compressor: An open-source Python library supporting popular model compression techniques."""
-__version__ = "3.4"
+__version__ = "3.4.1"