From 01cb2f25e3577b098dd142c7c37344277e678251 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 15 Jul 2021 14:29:49 +0200 Subject: [PATCH] LXMERT integration test typo (#12736) --- tests/test_modeling_lxmert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_lxmert.py b/tests/test_modeling_lxmert.py index 1febee25d5..624d3e5e5c 100644 --- a/tests/test_modeling_lxmert.py +++ b/tests/test_modeling_lxmert.py @@ -738,7 +738,7 @@ class LxmertModelIntegrationTest(unittest.TestCase): model = LxmertModel.from_pretrained(LXMERT_PRETRAINED_MODEL_ARCHIVE_LIST[0]) input_ids = torch.tensor([[101, 345, 232, 328, 740, 140, 1695, 69, 6078, 1588, 102]]) num_visual_features = 10 - _, visual_feats = np.random.seed(0), np.random.rand(1, num_visual_features, LxmertModel.config.visual_feat_dim) + _, visual_feats = np.random.seed(0), np.random.rand(1, num_visual_features, model.config.visual_feat_dim) _, visual_pos = np.random.seed(0), np.random.rand(1, num_visual_features, 4) visual_feats = torch.as_tensor(visual_feats, dtype=torch.float32) visual_pos = torch.as_tensor(visual_pos, dtype=torch.float32)