fix xlnet & transfotests

2020-03-08 16:25:03 +01:00
parent 66c827656f
commit b4a3a64744
4 changed files with 24 additions and 64 deletions
--- a/tests/test_modeling_tf_transfo_xl.py
+++ b/tests/test_modeling_tf_transfo_xl.py
@@ -519,20 +519,10 @@ class TFTransfoXLModelLanguageGenerationTest(unittest.TestCase):
            24,
            24,
            0,
-            29546,
-            40,
-            1092,
-            18,
-            8,
-            5854,
-            7,
-            1143,
-            2,
-            7,
+            33,
            1,
-            159,
-            99,
-            16,
+            1857,
+            2,
            1,
            1009,
            4,