From 506633f5ed2d4499e85284e7023a8c301794ebf0 Mon Sep 17 00:00:00 2001 From: oweller2 Date: Sat, 21 Jun 2025 23:48:58 -0400 Subject: [PATCH] update tests --- .../test_modeling_modernbert_decoder.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/models/modernbert_decoder/test_modeling_modernbert_decoder.py b/tests/models/modernbert_decoder/test_modeling_modernbert_decoder.py index 56436eb5935..9f0eb4657cb 100644 --- a/tests/models/modernbert_decoder/test_modeling_modernbert_decoder.py +++ b/tests/models/modernbert_decoder/test_modeling_modernbert_decoder.py @@ -19,7 +19,6 @@ from transformers import AutoTokenizer, ModernBertDecoderConfig, is_torch_availa from transformers.testing_utils import ( require_torch, slow, - torch_device, ) from ...causal_lm_tester import CausalLMModelTest, CausalLMModelTester @@ -131,9 +130,7 @@ class ModernBertDecoderIntegrationTest(unittest.TestCase): if version.parse(torch.__version__) < version.parse("2.4.0"): self.skipTest(reason="This test requires torch >= 2.4 to run.") - model = ModernBertDecoderForCausalLM.from_pretrained( - "blab-jhu/test-32m-dec", device_map=torch_device, torch_dtype=torch.float16 - ) + model = ModernBertDecoderForCausalLM.from_pretrained("blab-jhu/test-32m-dec") tokenizer = AutoTokenizer.from_pretrained("blab-jhu/test-32m-dec") # Create a longer input to test sliding window attention @@ -153,9 +150,7 @@ class ModernBertDecoderIntegrationTest(unittest.TestCase): if version.parse(torch.__version__) < version.parse("2.4.0"): self.skipTest(reason="This test requires torch >= 2.4 to run.") - model = ModernBertDecoderForSequenceClassification.from_pretrained( - "blab-jhu/test-32m-dec", num_labels=2, device_map=torch_device, torch_dtype=torch.float16 - ) + model = ModernBertDecoderForSequenceClassification.from_pretrained("blab-jhu/test-32m-dec", num_labels=2) tokenizer = AutoTokenizer.from_pretrained("blab-jhu/test-32m-dec") # Test with sample input