@@ -449,7 +449,8 @@ node = onnx.helper.make_node(
449
449
outputs = [' indices' , ' output' ],
450
450
mapping_file_name = ' vocabulary.txt' ,
451
451
unmapping_value = " unknown_word" ,
452
- model = model
452
+ model = model,
453
+ domain = ' ai.onnx.contrib'
453
454
)
454
455
455
456
inputs = np.array([" Hello world" , " Hello world louder" ], dtype = object ),
@@ -459,8 +460,8 @@ add_bos = np.array([0], dtype=np.bool_),
459
460
add_eos = np.array([0 ], dtype = np.bool_),
460
461
reverse = np.array([0 ], dtype = np.bool_)
461
462
462
- tokens = array([17486 , 1017 , 17486 , 1017 , 155 , 21869 ], dtype = int32)
463
- indices = array([0 , 2 , 6 ], dtype = int64)
463
+ tokens = np. array([17486 , 1017 , 17486 , 1017 , 155 , 21869 ], dtype = np. int32)
464
+ indices = np. array([0 , 2 , 6 ], dtype = np. int64)
464
465
465
466
expect(node, inputs = [inputs, nbest_size, alpha, add_bos, add_eos, reverse],
466
467
outputs = [tokens, indices], name = ' sp' )
@@ -1597,4 +1598,4 @@ def run_add_f():
1597
1598
}
1598
1599
out = sess.run(None, ort_inputs)[0]
1599
1600
` ` `
1600
- < /details>
1601
+ < /details>
0 commit comments