fix neural engine error (#1642)

LJ-underdog · web-flow · commit 816f475b1ec3 · 2024-06-28T18:56:06.000+08:00
Signed-off-by: intellinjun &lt;jun.lin@intel.com&gt;
diff --git a/examples/huggingface/pytorch/text-classification/deployment/emotion/distilbert_base_uncased/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/emotion/distilbert_base_uncased/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_base/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_base/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_base_cased/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_base_cased/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_base_cased/run_glue.py b/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_base_cased/run_glue.py
@@ -468,7 +468,7 @@ def preprocess_function(examples):
 
     # Get the metric function
     if data_args.task_name is not None:
-        metric = load_metric("glue", data_args.task_name)
+        metric = load_metric("glue", data_args.task_name,trust_remote_code=True)
     else:
         metric = load_metric("accuracy")
 
diff --git a/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_mini/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/mrpc/bert_mini/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/mrpc/distilbert_base_uncased/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/mrpc/distilbert_base_uncased/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/mrpc/roberta_base/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/mrpc/roberta_base/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/sparse/bert_mini/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/sparse/bert_mini/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/sparse/distilbert_base_uncased/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/sparse/distilbert_base_uncased/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/sst2/bert_mini/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/sst2/bert_mini/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/sst2/distilbert_base_uncased/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/sst2/distilbert_base_uncased/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
diff --git a/examples/huggingface/pytorch/text-classification/deployment/sst2/minilm_l6_h384_uncased/requirements.txt b/examples/huggingface/pytorch/text-classification/deployment/sst2/minilm_l6_h384_uncased/requirements.txt
@@ -1,4 +1,5 @@
-neural-compressor
+intel-extension-for-transformers==1.4.2
+neural-compressor==2.6
 transformers
 accelerate
 datasets >= 1.8.0
