Fix PT-TF equivalence test for GPT1 (#22586)

* Re-enable skipped test and fix the hidden state shape issue * Actually fix the bug instead of just doing something wrong
2023-04-05 13:16:00 +01:00
parent 0684284911
commit 2a91a9ef66
2 changed files with 7 additions and 5 deletions
--- a/tests/models/openai/test_modeling_openai.py
+++ b/tests/models/openai/test_modeling_openai.py
@@ -274,10 +274,6 @@ class OpenAIGPTModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTester
            model = OpenAIGPTModel.from_pretrained(model_name)
            self.assertIsNotNone(model)

-    @unittest.skip("Fix me Matt")
-    def test_pt_tf_model_equivalence(self):
-        pass
-

@require_torch
 class OPENAIGPTModelLanguageGenerationTest(unittest.TestCase):