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

5
atroposlib/envs/server_handling/server_manager.py
View file

@ -13,6 +13,7 @@ from atroposlib.envs.server_handling.server_baseline import (
ServerBaseline,
)
from atroposlib.envs.server_handling.server_harness import ServerHarness
from atroposlib.envs.server_handling.trl_vllm_server import TrlVllmServer
class ServerManagerConfig(BaseModel):
@ -36,11 +37,15 @@ class ServerManager:
if isinstance(configs, ServerBaseline):
if configs.server_type == "openai":
server_class = OpenAIServer
elif configs.server_type == "trl":
server_class = TrlVllmServer
else:
raise ValueError(f"Invalid server type: {configs.server_type}")
else:
if configs[0].server_type == "openai":
server_class = OpenAIServer
elif configs[0].server_type == "trl":
server_class = TrlVllmServer
else:
raise ValueError(f"Invalid server type: {configs[0].server_type}")
if testing: