get tokenizer from .env

This commit is contained in:
Shannon Sands 2026-02-03 14:50:37 +10:00
parent b5b1fef20a
commit 30221d8c20
8 changed files with 97 additions and 6 deletions

View file

@ -55,6 +55,7 @@ class HermesCompatTestEnvConfig(AgentEnvConfig):
description="Base URL for an OpenAI-compatible chat server (without /v1).",
)
server_model: str = Field(default="hermes-4-36b", description="Model name")
tokenizer_name: str = Field(default="NousResearch/Hermes-4.3-36B", description="Tokenizer name for RL tokenization")
class HermesCompatTestEnv(AgentEnv[HermesCompatTestEnvConfig]):
@ -83,7 +84,7 @@ class HermesCompatTestEnv(AgentEnv[HermesCompatTestEnvConfig]):
api_key = os.getenv("ATROPOS_SERVER_API_KEY") or os.getenv("OPENAI_API_KEY") or "local"
env_config = HermesCompatTestEnvConfig(
tokenizer_name="Qwen/Qwen2.5-1.5B-Instruct", # tokenization only
tokenizer_name=os.getenv("ATROPOS_TOKENIZER_NAME") or "NousResearch/Hermes-4.3-36B",
group_size=1,
use_wandb=False,
include_messages=True,