testLoadGPT2: Allow for variations depending on system

Quantization is imprecise, so we could get slightly different answers depending on the architecture.
smspillaz · Jul 27, 2023 · 60ceabe · 60ceabe
1 parent 2dd0d11
commit 60ceabe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/js/testLoadGPT2.js b/tests/js/testLoadGPT2.js
@@ -558,8 +558,8 @@ describe('GGML GPT2', function() {
       null
     );
 
-    expect(language_model.complete('The meaning of life is:', 7, null)).toEqual(
-      ['The meaning of life is: to live in a state of being', false]
+    expect(language_model.complete('The meaning of life is:', 7, null)[0]).toMatch(
+      /The meaning of life is\: to live in a state of (being|peace)/,
     );
   });
 })