From d81269d2b261d39967605e17a89b5688ebaedbd1 Mon Sep 17 00:00:00 2001 From: Sihan Chen <39623753+Spycsh@users.noreply.github.com> Date: Wed, 15 Nov 2023 16:31:54 +0800 Subject: [PATCH] remove unnecessary version limitation (#1373) --- examples/.config/model_params_tensorflow.json | 8 ++++---- .../quantization/ptq/smoothquant/requirements.txt | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/.config/model_params_tensorflow.json b/examples/.config/model_params_tensorflow.json index 7624fa89603..255410e7944 100644 --- a/examples/.config/model_params_tensorflow.json +++ b/examples/.config/model_params_tensorflow.json @@ -1776,28 +1776,28 @@ "gpt2_medium_sq": { "model_src_dir": "nlp/large_language_models/quantization/ptq/smoothquant", "dataset_location": "", - "input_model": "/tf_dataset2/models/tensorflow/gpt2-medium", + "input_model": "gpt2-medium", "main_script": "main.py", "batch_size": 16 }, "gpt2_medium": { "model_src_dir": "nlp/large_language_models/quantization/ptq/smoothquant", "dataset_location": "", - "input_model": "/tf_dataset2/models/tensorflow/gpt2-medium", + "input_model": "gpt2-medium", "main_script": "main.py", "batch_size": 16 }, "opt_125m_sq": { "model_src_dir": "nlp/large_language_models/quantization/ptq/smoothquant", "dataset_location": "", - "input_model": "/tf_dataset2/models/tensorflow/facebook-opt-125m", + "input_model": "facebook/opt-125m", "main_script": "main.py", "batch_size": 16 }, "opt_125m": { "model_src_dir": "nlp/large_language_models/quantization/ptq/smoothquant", "dataset_location": "", - "input_model": "/tf_dataset2/models/tensorflow/facebook-opt-125m", + "input_model": "facebook/opt-125m", "main_script": "main.py", "batch_size": 16 }, diff --git a/examples/tensorflow/nlp/large_language_models/quantization/ptq/smoothquant/requirements.txt b/examples/tensorflow/nlp/large_language_models/quantization/ptq/smoothquant/requirements.txt index 0994660e80c..fb574f91ff5 100644 --- a/examples/tensorflow/nlp/large_language_models/quantization/ptq/smoothquant/requirements.txt +++ b/examples/tensorflow/nlp/large_language_models/quantization/ptq/smoothquant/requirements.txt @@ -1,3 +1,3 @@ -tensorflow<=2.12.0 +tensorflow datasets -transformers<=4.29.2 \ No newline at end of file +transformers \ No newline at end of file