Răsfoiți Sursa

Add SentencePiece test file (#748)

Lutz Roeder 2 ani în urmă
părinte
comite
8c5ba0766e
2 a modificat fișierele cu 9 adăugiri și 2 ștergeri
  1. 2 2
      source/sentencepiece.js
  2. 7 0
      test/models.json

+ 2 - 2
source/sentencepiece.js

@@ -12,8 +12,8 @@ sentencepiece.ModelFactory = class {
             if (model &&
                 model['1'] && model['1']['1'] === 2 && model['1']['2'] === 5 && model['1']['3'] === 0 &&
                 model['2'] && model['2']['1'] === 2 && model['2']['2'] === 2 && model['2']['3'] === 0 &&
-                model['2']['4'] === 0 && model['2']['10'] === 5 && model['2']['16'] === 0 &&
-                model['2']['40'] === 0 && model['2']['41'] === 0 && model['2']['42'] === 0 && model['2']['43'] === 0) {
+                model['2']['4'] === 0 && model['2']['10'] === 5 &&
+                model['2']['40'] === 0 && model['2']['41'] === 0 && model['2']['42'] === 0) {
                 context.type = 'sentencepiece';
             }
         }

+ 7 - 0
test/models.json

@@ -6021,6 +6021,13 @@
     "assert":   [ "model.graphs[0].nodes[0].inputs[0].name == 'float32'" ],
     "link":     "https://github.com/lutzroeder/netron/issues/1113"
   },
+  {
+    "type":     "sentencepiece",
+    "target":   "bpe.model",
+    "source":   "https://github.com/lutzroeder/netron/files/14189544/bpe.model.zip[bpe.model]",
+    "format":   "SentencePiece",
+    "link":     "https://github.com/lutzroeder/netron/issues/748"
+  },
   {
     "type":     "sentencepiece",
     "target":   "spm_char.model",