Skip to content

Commit

Permalink
removing transformers script version checks since versions are now sy…
Browse files Browse the repository at this point in the history
…nced via pypi deps (#1553)
  • Loading branch information
bfineran committed May 10, 2023
1 parent f7f5275 commit 255e6c6
Show file tree
Hide file tree
Showing 4 changed files with 0 additions and 20 deletions.
5 changes: 0 additions & 5 deletions src/sparseml/transformers/masked_language_modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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",
Expand Down
5 changes: 0 additions & 5 deletions src/sparseml/transformers/question_answering.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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",
Expand Down
5 changes: 0 additions & 5 deletions src/sparseml/transformers/text_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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",
Expand Down
5 changes: 0 additions & 5 deletions src/sparseml/transformers/token_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,18 +48,13 @@
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
from sparseml.transformers.sparsification import Trainer, TrainingArguments
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",
Expand Down

0 comments on commit 255e6c6

Please sign in to comment.