|
<?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_391590" 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_391579" type="Const" version="opset1"> |
|
<data element_type="u8" shape="587404" offset="4" size="587404" /> |
|
<output> |
|
<port id="0" precision="U8"> |
|
<dim>587404</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="3" name="StringTensorUnpack_391582" type="StringTensorUnpack" version="extension"> |
|
<data mode="begins_ends" /> |
|
<input> |
|
<port id="0" precision="STRING"> |
|
<dim>-1</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="4" name="Constant_391584" type="Const" version="opset1"> |
|
<data element_type="u8" shape="7" offset="587408" size="7" /> |
|
<output> |
|
<port id="0" precision="U8"> |
|
<dim>7</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="5" name="Constant_391586" type="Const" version="opset1"> |
|
<data element_type="u8" shape="3" offset="587415" size="3" /> |
|
<output> |
|
<port id="0" precision="U8"> |
|
<dim>3</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="6" name="RegexNormalization_391587" type="RegexNormalization" version="extension"> |
|
<data global_replace="true" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="2" precision="U8"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="3" precision="U8"> |
|
<dim>7</dim> |
|
</port> |
|
<port id="4" precision="U8"> |
|
<dim>3</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="5" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="6" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="7" precision="U8"> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="7" name="StringTensorPack_391588" type="StringTensorPack" version="extension"> |
|
<data mode="begins_ends" /> |
|
<input> |
|
<port id="0" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="1" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="2" precision="U8"> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="3" precision="STRING"> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="8" name="SentencepieceTokenizer_391589" type="SentencepieceTokenizer" version="extension"> |
|
<data nbest_size="0" alpha="0" add_bos="true" add_eos="false" reverse="true" /> |
|
<input> |
|
<port id="0" precision="U8"> |
|
<dim>587404</dim> |
|
</port> |
|
<port id="1" precision="STRING"> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
<output> |
|
<port id="2" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>2</dim> |
|
</port> |
|
<port id="3" precision="I32"> |
|
<dim>-1</dim> |
|
</port> |
|
<port id="4" precision="I64"> |
|
<dim>2</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="9" name="Broadcast_391591" 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="10" name="Constant_391592" type="Const" version="opset1"> |
|
<data element_type="i32" shape="" offset="587418" size="4" /> |
|
<output> |
|
<port id="0" precision="I32" /> |
|
</output> |
|
</layer> |
|
<layer id="11" name="ShapeOf_391593" 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="12" name="Broadcast_391594" 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="13" name="ScatterNDUpdate_391598" 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="14" name="Constant_391602" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="587422" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="15" name="Reverse_391603" 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="16" name="Reverse_391603" 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="18" name="Constant_391599" type="Const" version="opset1"> |
|
<data element_type="i32" shape="" offset="0" size="4" /> |
|
<output> |
|
<port id="0" precision="I32" /> |
|
</output> |
|
</layer> |
|
<layer id="19" name="Broadcast_391600" 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="20" name="ScatterNDUpdate_391601" 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="21" name="Constant_391604" type="Const" version="opset1"> |
|
<data element_type="i64" shape="1" offset="587422" size="8" /> |
|
<output> |
|
<port id="0" precision="I64"> |
|
<dim>1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="22" name="Reverse_391605" 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="23" name="Reverse_391605" 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="24" name="Result_391606" type="Result" version="opset1"> |
|
<input> |
|
<port id="0" precision="I64"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</input> |
|
</layer> |
|
<layer id="17" name="Result_391607" 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="3" to-port="0" /> |
|
<edge from-layer="1" from-port="0" to-layer="9" to-port="0" /> |
|
<edge from-layer="2" from-port="0" to-layer="8" to-port="0" /> |
|
<edge from-layer="3" from-port="1" to-layer="6" to-port="0" /> |
|
<edge from-layer="3" from-port="3" to-layer="6" to-port="2" /> |
|
<edge from-layer="3" from-port="2" to-layer="6" to-port="1" /> |
|
<edge from-layer="4" from-port="0" to-layer="6" to-port="3" /> |
|
<edge from-layer="5" from-port="0" to-layer="6" to-port="4" /> |
|
<edge from-layer="6" from-port="5" to-layer="7" to-port="0" /> |
|
<edge from-layer="6" from-port="6" to-layer="7" to-port="1" /> |
|
<edge from-layer="6" from-port="7" to-layer="7" to-port="2" /> |
|
<edge from-layer="7" from-port="3" to-layer="8" to-port="1" /> |
|
<edge from-layer="8" from-port="4" to-layer="9" to-port="1" /> |
|
<edge from-layer="8" from-port="3" to-layer="11" to-port="0" /> |
|
<edge from-layer="8" from-port="3" to-layer="20" to-port="2" /> |
|
<edge from-layer="8" from-port="2" to-layer="20" to-port="1" /> |
|
<edge from-layer="8" from-port="2" to-layer="13" to-port="1" /> |
|
<edge from-layer="8" from-port="4" to-layer="19" to-port="1" /> |
|
<edge from-layer="9" from-port="2" to-layer="13" to-port="0" /> |
|
<edge from-layer="10" from-port="0" to-layer="12" to-port="0" /> |
|
<edge from-layer="11" from-port="1" to-layer="12" to-port="1" /> |
|
<edge from-layer="12" from-port="2" to-layer="13" to-port="2" /> |
|
<edge from-layer="13" from-port="3" to-layer="15" to-port="0" /> |
|
<edge from-layer="14" from-port="0" to-layer="15" to-port="1" /> |
|
<edge from-layer="15" from-port="2" to-layer="16" to-port="0" /> |
|
<edge from-layer="16" from-port="1" to-layer="17" to-port="0" /> |
|
<edge from-layer="18" from-port="0" to-layer="19" to-port="0" /> |
|
<edge from-layer="19" from-port="2" to-layer="20" to-port="0" /> |
|
<edge from-layer="20" from-port="3" to-layer="22" to-port="0" /> |
|
<edge from-layer="21" from-port="0" to-layer="22" to-port="1" /> |
|
<edge from-layer="22" from-port="2" to-layer="23" to-port="0" /> |
|
<edge from-layer="23" from-port="1" to-layer="24" to-port="0" /> |
|
</edges> |
|
<rt_info> |
|
<bos_token_id value="1" /> |
|
<chat_template value="{%- if messages[0]["role"] == "system" %} {%- set system_message = messages[0]["content"] %} {%- set loop_messages = messages[1:] %} {%- else %} {%- set loop_messages = messages %} {%- endif %} {%- if not tools is defined %} {%- set tools = none %} {%- endif %} {%- set user_messages = loop_messages | selectattr("role", "equalto", "user") | list %} {%- for message in loop_messages | rejectattr("role", "equalto", "tool") | rejectattr("role", "equalto", "tool_results") | selectattr("tool_calls", "undefined") %} {%- if (message["role"] == "user") != (loop.index0 % 2 == 0) %} {{- raise_exception("After the optional system message, conversation roles must alternate user/assistant/user/assistant/...") }} {%- endif %} {%- endfor %} {{- bos_token }} {%- for message in loop_messages %} {%- if message["role"] == "user" %} {%- if tools is not none and (message == user_messages[-1]) %} {{- "[AVAILABLE_TOOLS] [" }} {%- for tool in tools %} {%- set tool = tool.function %} {{- '{"type": "function", "function": {' }} {%- for key, val in tool.items() if key != "return" %} {%- if val is string %} {{- '"' + key + '": "' + val + '"' }} {%- else %} {{- '"' + key + '": ' + val|tojson }} {%- endif %} {%- if not loop.last %} {{- ", " }} {%- endif %} {%- endfor %} {{- "}}" }} {%- if not loop.last %} {{- ", " }} {%- else %} {{- "]" }} {%- endif %} {%- endfor %} {{- "[/AVAILABLE_TOOLS]" }} {%- endif %} {%- if loop.last and system_message is defined %} {{- "[INST] " + system_message + "\n\n" + message["content"] + "[/INST]" }} {%- else %} {{- "[INST] " + message["content"] + "[/INST]" }} {%- endif %} {%- elif message["role"] == "tool_calls" or message.tool_calls is defined %} {%- if message.tool_calls is defined %} {%- set tool_calls = message.tool_calls %} {%- else %} {%- set tool_calls = message.content %} {%- endif %} {{- "[TOOL_CALLS] [" }} {%- for tool_call in tool_calls %} {%- set out = tool_call.function|tojson %} {{- out[:-1] }} {%- if not tool_call.id is defined or tool_call.id|length != 9 %} {{- raise_exception("Tool call IDs should be alphanumeric strings with length 9!") }} {%- endif %} {{- ', "id": "' + tool_call.id + '"}' }} {%- if not loop.last %} {{- ", " }} {%- else %} {{- "]" + eos_token }} {%- endif %} {%- endfor %} {%- elif message["role"] == "assistant" %} {{- " " + message["content"] + eos_token}} {%- elif message["role"] == "tool_results" or message["role"] == "tool" %} {%- if message.content is defined and message.content.content is defined %} {%- set content = message.content.content %} {%- else %} {%- set content = message.content %} {%- endif %} {{- '[TOOL_RESULTS] {"content": ' + content|string + ", " }} {%- if not message.tool_call_id is defined or message.tool_call_id|length != 9 %} {{- raise_exception("Tool call IDs should be alphanumeric strings with length 9!") }} {%- endif %} {{- '"call_id": "' + message.tool_call_id + '"}[/TOOL_RESULTS]' }} {%- else %} {{- raise_exception("Only user and assistant roles are supported, with the exception of an initial optional system message!") }} {%- endif %} {%- endfor %} " /> |
|
<eos_token_id value="2" /> |
|
<original_tokenizer_class value="<class 'transformers.models.llama.tokenization_llama_fast.LlamaTokenizerFast'>" /> |
|
</rt_info> |
|
</net> |
|
|
