This repository was archived by the owner on Oct 25, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +7
-3
lines changed
examples/huggingface/pytorch/summarization/quantization
intel_extension_for_transformers/backends/neural_engine/test/pytest Expand file tree Collapse file tree 3 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -72,9 +72,11 @@ function run_benchmark {
7272 elif [ " ${topology} " == " t5_base_cnn_dynamic" ]; then
7373 DATASET_NAME=" cnn_dailymail"
7474 model_name_or_path=" flax-community/t5-base-cnn-dm"
75+ pip install transformers==4.26.0
7576 elif [ " ${topology} " == " t5_large_cnn_dynamic" ]; then
7677 DATASET_NAME=" cnn_dailymail"
7778 model_name_or_path=" sysresearch101/t5-large-finetuned-xsum-cnn"
79+ pip install transformers==4.26.0
7880 elif [ " ${topology} " == " flan_t5_large_samsum_dynamic" ]; then
7981 DATASET_NAME=" samsum"
8082 model_name_or_path=" stacked-summaries/flan-t5-large-stacked-samsum-1024"
Original file line number Diff line number Diff line change @@ -50,10 +50,12 @@ function run_tuning {
5050 DATASET_NAME=" cnn_dailymail"
5151 model_name_or_path=" flax-community/t5-base-cnn-dm"
5252 approach=" PostTrainingDynamic"
53+ pip install transformers==4.26.0
5354 elif [ " ${topology} " == " t5_large_cnn_dynamic" ]; then
5455 DATASET_NAME=" cnn_dailymail"
5556 model_name_or_path=" sysresearch101/t5-large-finetuned-xsum-cnn"
5657 approach=" PostTrainingDynamic"
58+ pip install transformers==4.26.0
5759 elif [ " ${topology} " == " flan_t5_large_samsum_dynamic" ]; then
5860 DATASET_NAME=" samsum"
5961 model_name_or_path=" stacked-summaries/flan-t5-large-stacked-samsum-1024"
Original file line number Diff line number Diff line change @@ -3,6 +3,6 @@ onnxruntime
33datasets
44transformers
55onnxruntime_extensions
6- torch == 1.13.1
7- intel-tensorflow == 2.10.0
8- intel-extension-for-pytorch == 1.13 .0
6+ torch == 2.0
7+ intel-tensorflow
8+ intel-extension-for-pytorch == 2.0 .0
You can’t perform that action at this time.
0 commit comments