We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bb57589 commit 4410e90Copy full SHA for 4410e90
apps/grpo/main.py
@@ -266,7 +266,12 @@ async def sample(self) -> dict[str, str] | None:
266
267
@endpoint
268
async def pad_token(self):
269
- return self._tokenizer.pad_token_id
+ # 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
275
276
277
async def drop_weights(version: int):
0 commit comments