Remove old code for PyTorch, Accelerator and tokenizers (#37234)
* Remove unneeded library version checks Signed-off-by: cyy <cyyever@outlook.com> * Remove PyTorch condition Signed-off-by: cyy <cyyever@outlook.com> * Remove PyTorch condition Signed-off-by: cyy <cyyever@outlook.com> * Fix ROCm get_device_capability Signed-off-by: cyy <cyyever@outlook.com> * Revert "Fix ROCm get_device_capability" This reverts commit 0e756434bd7e74ffd73de5500476072b096570a6. * Remove unnecessary check Signed-off-by: cyy <cyyever@outlook.com> * Revert changes Signed-off-by: cyy <cyyever@outlook.com> --------- Signed-off-by: cyy <cyyever@outlook.com>
This commit is contained in:
@@ -81,7 +81,6 @@ from transformers.utils.import_utils import (
|
||||
is_tf_available,
|
||||
is_torch_npu_available,
|
||||
is_torch_sdpa_available,
|
||||
is_torchdynamo_available,
|
||||
)
|
||||
|
||||
|
||||
@@ -1483,8 +1482,6 @@ class ModelUtilsTest(TestCasePlus):
|
||||
model.warn_if_padding_and_no_attention_mask(input_ids, attention_mask=None)
|
||||
self.assertIn("You may ignore this warning if your `pad_token_id`", cl.out)
|
||||
|
||||
if not is_torchdynamo_available():
|
||||
self.skipTest(reason="torchdynamo is not available")
|
||||
with self.subTest("Ensure that the warning code is skipped when compiling with torchdynamo."):
|
||||
logger.warning_once.cache_clear()
|
||||
from torch._dynamo import config, testing
|
||||
|
||||
Reference in New Issue
Block a user