From 845aa832b7637e158a6aa1baae40f0c218e90824 Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Wed, 8 Nov 2023 09:07:32 -0500 Subject: [PATCH] Remove unused param from example script tests (#27354) Unused param --- examples/pytorch/test_accelerate_examples.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/examples/pytorch/test_accelerate_examples.py b/examples/pytorch/test_accelerate_examples.py index 22dd7dea27..cdc9d6dc5b 100644 --- a/examples/pytorch/test_accelerate_examples.py +++ b/examples/pytorch/test_accelerate_examples.py @@ -28,7 +28,6 @@ from accelerate.utils import write_basic_config from transformers.testing_utils import ( TestCasePlus, backend_device_count, - is_torch_fp16_available_on_device, run_command, slow, torch_device, @@ -93,9 +92,6 @@ class ExamplesTestsNoTrainer(TestCasePlus): --with_tracking """.split() - if is_torch_fp16_available_on_device(torch_device): - testargs.append("--fp16") - run_command(self._launch_args + testargs) result = get_results(tmp_dir) self.assertGreaterEqual(result["eval_accuracy"], 0.75) @@ -325,9 +321,6 @@ class ExamplesTestsNoTrainer(TestCasePlus): --checkpointing_steps 1 """.split() - if is_torch_fp16_available_on_device(torch_device): - testargs.append("--fp16") - run_command(self._launch_args + testargs) result = get_results(tmp_dir) # The base model scores a 25%