PyRIT
PyRIT copied to clipboard
FEAT: Remove dependency on fastchat for model conversation templates
Replace FastChat’s custom prompt logic with Hugging Face’s apply_chat_template() and tokenizer_config.json.
- Enable use_fast = True in the tokenizer.
- Update _update_ids() in
pyrit/auxiliary_attacks/gcg/attack/base/attack_manager.pyto call apply_chat_template(), generating model-specific prompts according to the conversation template.
@romanlutz