Aryan Khandhadiya

Improve error message for invalid virtualenvs.prompt config

4

# Pull Request Check List Resolves: #Resolves: #10635 ### Summary Improve error message when an invalid template variable is used in `virtualenvs.prompt` configuration. ### Details Previously an invalid placeholder such...

Fix: Pass missing KV cache constants to generation loop (#1036)

1

Fixes #1036. Currently, the memory management constants MAX_KV_SIZE and KEEP_KV_SIZE are defined in constants.py but are never passed to the make_kv_cache function in generate.py. This results in the model defaulting...

feat(functional_jax): support reset options for JAX envs

2

# Description This PR adds support for `options` in `reset()` for functional JAX environments.Currently, `FunctionalJaxEnv` and `FunctionalJaxVectorEnv` accept an `options` argument in `reset()` but ignore it. This makes reset-time configuration...