NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add trl server

Browse source 
add gsm8k example for axolotl checking
This commit is contained in:
dmahan93 2025-05-12 19:04:46 -05:00
parent 96be544228
commit 0aaf59fc9a
4 changed files with 587 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
atroposlib/envs/server_handling/server_baseline.py
View file

@ -35,7 +35,7 @@ class APIServerConfig(ServerBaseline):
api_key: Optional[str] = Field(default="", description="API key for the server.")
base_url: Optional[str] = Field(default="", description="Base URL for the server.")
server_type: Literal["openai"] = Field(
server_type: Literal["openai", "trl"] = Field(
default="openai", description="Type of server to use, openai or trl"
)
n_kwarg_is_ignored: bool = Field(