From 3b1185491109061e8582774d173d8203245ad464 Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Wed, 10 May 2023 10:53:24 -0400 Subject: [PATCH] removing transformers script version checks since versions are now synced via pypi deps (#1553) --- src/sparseml/transformers/masked_language_modeling.py | 5 ----- src/sparseml/transformers/question_answering.py | 5 ----- src/sparseml/transformers/text_classification.py | 5 ----- src/sparseml/transformers/token_classification.py | 5 ----- 4 files changed, 20 deletions(-) diff --git a/src/sparseml/transformers/masked_language_modeling.py b/src/sparseml/transformers/masked_language_modeling.py index a533e9c0a17..d2c1118fe5a 100644 --- a/src/sparseml/transformers/masked_language_modeling.py +++ b/src/sparseml/transformers/masked_language_modeling.py @@ -51,7 +51,6 @@ set_seed, ) from transformers.trainer_utils import get_last_checkpoint -from transformers.utils import check_min_version from transformers.utils.versions import require_version from sparseml.pytorch.utils.distributed import record @@ -66,10 +65,6 @@ ] -# Will error if the minimal version of Transformers is not installed. -# Remove at your own risks -check_min_version("4.18.0.dev0") - require_version( "datasets>=1.18.0", "To fix: pip install -r examples/pytorch/language-modeling/requirements.txt", diff --git a/src/sparseml/transformers/question_answering.py b/src/sparseml/transformers/question_answering.py index a975f051320..32cfbdb479d 100644 --- a/src/sparseml/transformers/question_answering.py +++ b/src/sparseml/transformers/question_answering.py @@ -45,7 +45,6 @@ set_seed, ) from transformers.trainer_utils import get_last_checkpoint -from transformers.utils import check_min_version from transformers.utils.versions import require_version from sparseml.pytorch.utils.distributed import record @@ -57,10 +56,6 @@ from sparseml.transformers.utils import SparseAutoModel, get_shared_tokenizer_src -# Will error if the minimal version of Transformers is not installed. Remove at your -# own risks. -check_min_version("4.18.0.dev0") - require_version( "datasets>=1.18.0", "To fix: pip install -r examples/pytorch/question-answering/requirements.txt", diff --git a/src/sparseml/transformers/text_classification.py b/src/sparseml/transformers/text_classification.py index f45a31f42c9..7ad300712fe 100644 --- a/src/sparseml/transformers/text_classification.py +++ b/src/sparseml/transformers/text_classification.py @@ -49,7 +49,6 @@ set_seed, ) from transformers.trainer_utils import get_last_checkpoint -from transformers.utils import check_min_version from transformers.utils.versions import require_version from sparseml.pytorch.utils.distributed import record @@ -61,10 +60,6 @@ ) -# Will error if the minimal version of Transformers is not installed. -# Remove at your own risks. -check_min_version("4.18.0.dev0") - require_version( "datasets>=1.18.0", "To fix: pip install -r examples/pytorch/text-classification/requirements.txt", diff --git a/src/sparseml/transformers/token_classification.py b/src/sparseml/transformers/token_classification.py index 7a6db557b31..34504b8f60d 100644 --- a/src/sparseml/transformers/token_classification.py +++ b/src/sparseml/transformers/token_classification.py @@ -48,7 +48,6 @@ set_seed, ) from transformers.trainer_utils import get_last_checkpoint -from transformers.utils import check_min_version from transformers.utils.versions import require_version from sparseml.pytorch.utils.distributed import record @@ -56,10 +55,6 @@ from sparseml.transformers.utils import SparseAutoModel, get_shared_tokenizer_src -# Will error if the minimal version of Transformers is not installed. -# Remove at your own risks -check_min_version("4.18.0.dev0") - require_version( "datasets>=1.18.0", "To fix: pip install -r examples/pytorch/token-classification/requirements.txt",