Purge unused ModelTester code (#37085)

* Purge correctly this time

* Remove more methods from recent PRs

* make fixup
This commit is contained in:
Matt
2025-04-03 17:48:35 +01:00
committed by GitHub
parent 1b29409d89
commit 2d46a08b63
72 changed files with 3 additions and 4286 deletions

View File

@@ -75,17 +75,6 @@ class TimmWrapperModelTester:
def get_config(self):
return TimmWrapperConfig.from_pretrained(self.model_name)
def create_and_check_model(self, config, pixel_values):
model = TimmWrapperModel(config=config)
model.to(torch_device)
model.eval()
with torch.no_grad():
result = model(pixel_values)
self.parent.assertEqual(
result.feature_map[-1].shape,
(self.batch_size, model.channels[-1], 14, 14),
)
def prepare_config_and_inputs_for_common(self):
config_and_inputs = self.prepare_config_and_inputs()
config, pixel_values = config_and_inputs