Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/llmart/attack.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,8 +290,8 @@ def train(
pbar.set_postfix(postfix)

# Save tokens with highest success rate
if success_rate > best_success_rate:
best_success_rate, best_attack = loss, copy.deepcopy(attack)
if success_rate >= best_success_rate:
best_success_rate, best_attack = success_rate, copy.deepcopy(attack)

# Exit attack loop if we found a successful attack across all training examples
if (
Expand Down