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.