Skip to content

Commit 256b243

Browse files
committed
add "token_logits_over_infill" in attr
1 parent 4ea4861 commit 256b243

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

protein_tune_rl/metrics/iglm_tokenwise_kl.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ def _fetch_logits_over_infill(
127127
"""
128128
# Try a few attr names to keep this robust to minor refactors
129129
for attr in (
130+
"token_logits_over_infill", # preferred if available
130131
"logits",
131132
"get_logits",
132133
"token_logits",

0 commit comments

Comments
 (0)