Skip to content

Commit 8f529c8

Browse files
fixing config
1 parent 26619fd commit 8f529c8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

recipes/ppo_full_finetune_single_device.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,11 @@ def setup(self, cfg: DictConfig) -> None:
234234

235235
# setup a context manager for enabling KV-cacheing during
236236
# trajectory generation if enabled in the config
237+
if self.enable_kv_cache and self._tokenizer.max_seq_len is None:
238+
raise ValueError(
239+
"`tokenizer.max_seq_len` is required when `enable_kv_cache` is `True`. "
240+
"Please set `max_seq_len` in tokenizer config."
241+
)
237242
self.cache_ctx_manager = lambda enable_kv_cache: (
238243
local_kv_cache(
239244
self._policy_model,

0 commit comments

Comments
 (0)