Skip to content

Commit 4a6188e

Browse files
committed
[bugfix] fix no weight_version for aborted samples
1 parent d22225f commit 4a6188e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

slime/rollout/sglang_rollout.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,8 @@ async def generate(args, sample: Sample, sampling_params) -> Sample:
116116
sample.tokens = sample.tokens + new_response_tokens
117117
sample.response_length += len(new_response_tokens)
118118
sample.response += output["text"]
119-
sample.weight_versions.append(output["meta_info"]["weight_version"])
119+
if "weight_version" in output["meta_info"]:
120+
sample.weight_versions.append(output["meta_info"]["weight_version"])
120121
if sample.rollout_log_probs is None:
121122
sample.rollout_log_probs = []
122123
sample.rollout_log_probs += new_response_log_probs

0 commit comments

Comments
 (0)