README.md

visual-tool

To run:

python main.py

To add support for a new model, you need to implement your own chat_template and possibly need to change the create_prob_ls to fit your tokenizer style.