(
messages: List[llama_types.ChatCompletionRequestMessage],
**kwargs: Any,
)
| 1199 | |
| 1200 | @register_chat_format("redpajama-incite") |
| 1201 | def format_redpajama_incite( |
| 1202 | messages: List[llama_types.ChatCompletionRequestMessage], |
| 1203 | **kwargs: Any, |
| 1204 | ) -> ChatFormatterResponse: |
| 1205 | _system_message = _get_system_message(messages) |
| 1206 | _roles = dict(user="<human>", assistant="<bot>") |
| 1207 | _sep = "\n" |
| 1208 | _stop = "<human>" |
| 1209 | system_message = _system_message |
| 1210 | _messages = _map_roles(messages, _roles) |
| 1211 | _messages.append((_roles["assistant"], None)) |
| 1212 | _prompt = _format_add_colon_single(system_message, _messages, _sep) |
| 1213 | return ChatFormatterResponse(prompt=_prompt, stop=_stop) |
| 1214 | |
| 1215 | |
| 1216 | @register_chat_format("snoozy") |
nothing calls this directly
no test coverage detected
searching dependent graphs…