Skip to content

Commit 4410e90

Browse files
authored
use eos_token_id if there is no pad_token_id to avoid sending None (#541)
1 parent bb57589 commit 4410e90

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

apps/grpo/main.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,12 @@ async def sample(self) -> dict[str, str] | None:
266266

267267
@endpoint
268268
async def pad_token(self):
269-
return self._tokenizer.pad_token_id
269+
# Use pad_token_id if available, otherwise use eos_token_id
270+
# Llama models don't have a pad token by default
271+
if self._tokenizer.pad_token_id is not None:
272+
return self._tokenizer.pad_token_id
273+
else:
274+
return self._tokenizer.eos_token_id
270275

271276

272277
async def drop_weights(version: int):

0 commit comments

Comments
 (0)