pychatml 0.0.4

A package for converting chat messages to and from ChatML format

pychatml

The pychatml package allows you to convert chat interfaces from and to the ChatML format.

Installation

You can install the pychatml package using pip:

pip install pychatml

What

Makes it easy to integrate between different chat formats and models.

import pychatml

PROMPT = """[INST Hi, how are you? [/INST] Good thanks! 
[INST] Can you help me with this math program? [/INST]"""

pychatml.llama2.to_chatml(PROMPT)

[
    {"role": "user", "content": "Hi, how are you?"},
    {"role": "assistant", "content": "Good thanks!"},
    {"role": "user", "content": "Can you help me with this math program?"},
]

Supported formats

• Llama 2
• Anthropic
• Alpaca
• Vicuna/ShareGPT

Why?