SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Revert 22152 MaskedImageCompletionOutput changes (#22187)

Browse Source 
Revert changes
This commit is contained in:
amyeroberts
2023-03-15 17:37:23 +00:00
committed by GitHub
parent 7b0e2cfdfb
commit 737681477c
3 changed files with 7 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/vit/test_modeling_vit.py
View File

@@ -134,7 +134,7 @@ class ViTModelTester:
model.eval()
result = model(pixel_values)
self.parent.assertEqual(
result.reconstruction.shape, (self.batch_size, self.num_channels, self.image_size, self.image_size)
result.logits.shape, (self.batch_size, self.num_channels, self.image_size, self.image_size)
)
# test greyscale images
@@ -145,7 +145,7 @@ class ViTModelTester:
pixel_values = floats_tensor([self.batch_size, 1, self.image_size, self.image_size])
result = model(pixel_values)
self.parent.assertEqual(result.reconstruction.shape, (self.batch_size, 1, self.image_size, self.image_size))
self.parent.assertEqual(result.logits.shape, (self.batch_size, 1, self.image_size, self.image_size))
def create_and_check_for_image_classification(self, config, pixel_values, labels):
config.num_labels = self.type_sequence_label_size