fix

deepjavalibrary · Jul 27, 2023 · 7a4e63f · 7a4e63f
1 parent a8772c7
commit 7a4e63f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/...runtime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/engine/GptTranslatorTest.java b/...runtime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/engine/GptTranslatorTest.java
@@ -85,7 +85,6 @@ public void testGpt2() throws TranslateException, ModelException, IOException {
             Assert.assertEquals(nextTokenId, 257);
             NDList list = res.getPastKeyValuesList();
             Assert.assertEquals(list.size(), 24);
-            Assert.assertEquals(res.getHiddenState().getShape().get(0), 1);
         }
     }
 }