This commit is contained in:
Ilyas Moutawwakil
2025-06-23 10:56:51 +02:00
committed by GitHub
parent 2166b6b4ff
commit 984ff89e73
16 changed files with 618 additions and 14 deletions

View File

@@ -58,6 +58,7 @@ from transformers.testing_utils import (
is_staging_test,
require_accelerate,
require_flax,
require_non_hpu,
require_read_token,
require_safetensors,
require_tf,
@@ -1002,6 +1003,7 @@ class ModelUtilsTest(TestCasePlus):
self.assertIsNotNone(model)
@require_non_hpu
@require_accelerate
@mark.accelerate_tests
@require_torch_multi_accelerator