chore: fix typos in the tests directory (#36813)
* chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * chore: fix typos in the tests * fix: format codes * chore: fix copy mismatch issue * fix: format codes * chore: fix copy mismatch issue * chore: fix copy mismatch issue * chore: fix copy mismatch issue * chore: restore previous words * chore: revert unexpected changes
This commit is contained in:
@@ -497,7 +497,7 @@ class LayoutLMv2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
|
||||
single_batch_shape = value.shape[0] // batch_size
|
||||
single_row_input[key] = value[:single_batch_shape]
|
||||
elif hasattr(value, "tensor"):
|
||||
# layoutlmv2uses ImageList intead of pixel values (needs for torchscript)
|
||||
# layoutlmv2uses ImageList instead of pixel values (needs for torchscript)
|
||||
single_row_input[key] = value.tensor[:single_batch_shape]
|
||||
|
||||
with torch.no_grad():
|
||||
|
||||
Reference in New Issue
Block a user