# train the detector
torchrun --nproc_per_node=8 training_stage_one.py --config configs/train_configs/llama2-7b-chat-4.json
# test TIM
python test_stage.py --config configs/eval_configs/I-FSJ/llama2-7b-chat-0.json