|
<?xml version="1.0"?> |
|
<net name="tokenizer" version="11"> |
|
<layers> |
|
<layer id="0" name="string_input" type="Parameter" version="opset1"> |
|
<data shape="?" element_type="string" /> |
|
<output> |
|
<port id="0" precision="STRING" names="string_input"> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="1" name="Constant_273721" type="Const" version="opset1"> |
|
<data element_type="i32" shape="" offset="0" size="4" /> |
|
<output> |
|
<port id="0" precision="I32" /> |
|
</output> |
|
</layer> |
|
<layer id="2" name="Constant_273713" type="Const" version="opset1"> |
|
<data element_type="u8" shape="499969" offset="4" size="499969" /> |
|
<output> |
|
<port id="0" precision="U8"> |
|
<dim>499969</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="3" name="Constant_273717" type="Const" version="opset1"> |
|
<data element_type="u8" shape="223" offset="499973" size="223" /> |
|
<output> |
|
<port id="0" precision="U8"> |
|
<dim>223</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="4" name="StringTensorUnpack_273718" type="StringTensorUnpack" version="extension"> |
|
<data mode="begins_ends" /> |
|
<input> |
|
<port id="0" precision="U8"> |
|
<dim>223</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="1" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="3" precision="U8"> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="5" name="Constant_273719" type="Const" version="opset1"> |
|
<data element_type="i32" shape="14" offset="500196" size="56" /> |
|
<output> |
|
<port id="0" precision="I32"> |
|
<dim>14</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="6" name="SentencepieceTokenizer_273720" type="SentencepieceTokenizer" version="extension"> |
|
<data nbest_size="0" alpha="0" add_bos="false" add_eos="false" reverse="true" /> |
|
<input> |
|
<port id="0" precision="U8"> |
|
<dim>499969</dim> |
|
</port> |
|
<port id="1" precision="STRING"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="3" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="4" precision="U8"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="5" precision="I32"> |
|
<dim>14</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="6" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>2</dim> |
|
</port> |
|
<port id="7" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="8" precision="I64"> |
|
<dim>2</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="7" name="Broadcast_273722" type="Broadcast" version="opset3"> |
|
<data mode="numpy" /> |
|
<input> |
|
<port id="0" precision="I32" /> |
|
<port id="1" precision="I64"> |
|
<dim>2</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="8" name="Constant_273723" type="Const" version="opset1"> |
|
<data element_type="i32" shape="" offset="500252" size="4" /> |
|
<output> |
|
<port id="0" precision="I32" /> |
|
</output> |
|
</layer> |
|
<layer id="9" name="ShapeOf_273724" type="ShapeOf" version="opset3"> |
|
<data output_type="i64" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="1" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="10" name="Broadcast_273725" type="Broadcast" version="opset3"> |
|
<data mode="numpy" /> |
|
<input> |
|
<port id="0" precision="I32" /> |
|
<port id="1" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="11" name="ScatterNDUpdate_273729" type="ScatterNDUpdate" version="opset4"> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>2</dim> |
|
</port> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="3" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="12" name="Constant_273733" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500256" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="13" name="Reverse_273734" type="Reverse" version="opset1"> |
|
<data mode="index" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="14" name="Constant_273742" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500264" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="15" name="Constant_273743" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500272" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="16" name="Constant_273744" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500280" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="17" name="Constant_273745" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500256" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="18" name="Slice_273746" type="Slice" version="opset8"> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
<port id="2" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
<port id="3" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
<port id="4" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="5" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="19" name="Slice_273746" type="Convert" version="opset1"> |
|
<data destination_type="i64" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="1" precision="I64" names="attention_mask"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="21" name="Constant_273730" type="Const" version="opset1"> |
|
<data element_type="i32" shape="" offset="500288" size="4" /> |
|
<output> |
|
<port id="0" precision="I32" /> |
|
</output> |
|
</layer> |
|
<layer id="22" name="Broadcast_273731" type="Broadcast" version="opset3"> |
|
<data mode="bidirectional" /> |
|
<input> |
|
<port id="0" precision="I32" /> |
|
<port id="1" precision="I64"> |
|
<dim>2</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="23" name="ScatterNDUpdate_273732" type="ScatterNDUpdate" version="opset4"> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>2</dim> |
|
</port> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="3" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="24" name="Constant_273735" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500256" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="25" name="Reverse_273736" type="Reverse" version="opset1"> |
|
<data mode="index" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="2" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="26" name="Constant_273737" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500264" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="27" name="Constant_273738" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500272" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="28" name="Constant_273739" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500280" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="29" name="Constant_273740" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="500256" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="30" name="Slice_273741" type="Slice" version="opset8"> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
<port id="2" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
<port id="3" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
<port id="4" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="5" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="31" name="Slice_273741" type="Convert" version="opset1"> |
|
<data destination_type="i64" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="1" precision="I64" names="input_ids"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="32" name="Result_273747" type="Result" version="opset1"> |
|
<input> |
|
<port id="0" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
</layer> |
|
<layer id="20" name="Result_273748" type="Result" version="opset1"> |
|
<input> |
|
<port id="0" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
</layer> |
|
</layers> |
|
<edges> |
|
<edge from-layer="0" from-port="0" to-layer="6" to-port="1" /> |
|
<edge from-layer="1" from-port="0" to-layer="7" to-port="0" /> |
|
<edge from-layer="2" from-port="0" to-layer="6" to-port="0" /> |
|
<edge from-layer="3" from-port="0" to-layer="4" to-port="0" /> |
|
<edge from-layer="4" from-port="1" to-layer="6" to-port="2" /> |
|
<edge from-layer="4" from-port="2" to-layer="6" to-port="3" /> |
|
<edge from-layer="4" from-port="3" to-layer="6" to-port="4" /> |
|
<edge from-layer="5" from-port="0" to-layer="6" to-port="5" /> |
|
<edge from-layer="6" from-port="8" to-layer="22" to-port="1" /> |
|
<edge from-layer="6" from-port="6" to-layer="23" to-port="1" /> |
|
<edge from-layer="6" from-port="7" to-layer="23" to-port="2" /> |
|
<edge from-layer="6" from-port="6" to-layer="11" to-port="1" /> |
|
<edge from-layer="6" from-port="7" to-layer="9" to-port="0" /> |
|
<edge from-layer="6" from-port="8" to-layer="7" to-port="1" /> |
|
<edge from-layer="7" from-port="2" to-layer="11" to-port="0" /> |
|
<edge from-layer="8" from-port="0" to-layer="10" to-port="0" /> |
|
<edge from-layer="9" from-port="1" to-layer="10" to-port="1" /> |
|
<edge from-layer="10" from-port="2" to-layer="11" to-port="2" /> |
|
<edge from-layer="11" from-port="3" to-layer="13" to-port="0" /> |
|
<edge from-layer="12" from-port="0" to-layer="13" to-port="1" /> |
|
<edge from-layer="13" from-port="2" to-layer="18" to-port="0" /> |
|
<edge from-layer="14" from-port="0" to-layer="18" to-port="1" /> |
|
<edge from-layer="15" from-port="0" to-layer="18" to-port="2" /> |
|
<edge from-layer="16" from-port="0" to-layer="18" to-port="3" /> |
|
<edge from-layer="17" from-port="0" to-layer="18" to-port="4" /> |
|
<edge from-layer="18" from-port="5" to-layer="19" to-port="0" /> |
|
<edge from-layer="19" from-port="1" to-layer="20" to-port="0" /> |
|
<edge from-layer="21" from-port="0" to-layer="22" to-port="0" /> |
|
<edge from-layer="22" from-port="2" to-layer="23" to-port="0" /> |
|
<edge from-layer="23" from-port="3" to-layer="25" to-port="0" /> |
|
<edge from-layer="24" from-port="0" to-layer="25" to-port="1" /> |
|
<edge from-layer="25" from-port="2" to-layer="30" to-port="0" /> |
|
<edge from-layer="26" from-port="0" to-layer="30" to-port="1" /> |
|
<edge from-layer="27" from-port="0" to-layer="30" to-port="2" /> |
|
<edge from-layer="28" from-port="0" to-layer="30" to-port="3" /> |
|
<edge from-layer="29" from-port="0" to-layer="30" to-port="4" /> |
|
<edge from-layer="30" from-port="5" to-layer="31" to-port="0" /> |
|
<edge from-layer="31" from-port="1" to-layer="32" to-port="0" /> |
|
</edges> |
|
<rt_info> |
|
<bos_token_id value="1" /> |
|
<chat_template value="{% for message in messages %}{% if message['role'] == 'system' %}{{'<|system|> ' + message['content'] + '<|end|> '}}{% elif message['role'] == 'user' %}{{'<|user|> ' + message['content'] + '<|end|> '}}{% elif message['role'] == 'assistant' %}{{'<|assistant|> ' + message['content'] + '<|end|> '}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|> ' }}{% else %}{{ eos_token }}{% endif %}" /> |
|
<eos_token_id value="32000" /> |
|
<original_tokenizer_class value="<class 'transformers.models.llama.tokenization_llama_fast.LlamaTokenizerFast'>" /> |
|
<pad_token_id value="32000" /> |
|
</rt_info> |
|
</net> |
|
|
