Skip to content

Commit fd8047a

Browse files
committed
Merge branch 'debug/huoshan_zirui' into debug/vllm_logp_lpx
2 parents 4decd59 + 2d2a31f commit fd8047a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

rollouter/model_service.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -874,6 +874,9 @@ async def generate(self, messages: List[Dict[str, Any]], **kwargs) -> str:
874874
raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e))
875875

876876
async def save(self, messages: List[Dict], reward: float, task_id: str, trace_id: str):
877+
if not self.save_local:
878+
return {"status": "skipped"}
879+
877880
try:
878881
save_dir = os.path.join(os.getcwd(), f"{task_id}_trace-{trace_id}")
879882
os.makedirs(save_dir, exist_ok=True)

0 commit comments

Comments
 (0)