Unify forward output datastructure (#11124)

2025-10-03 00:28:57 +08:00
parent 3511b37099
commit 458611de77
12 changed files with 180 additions and 135 deletions
@@ -173,8 +173,7 @@ class SchedulerOutputProcessorMixin:
            self.set_next_batch_sampling_info_done(batch)

        else:  # embedding or reward model
-            embeddings, bid = result.embeddings, result.bid
-            embeddings = embeddings.tolist()
+            embeddings = result.embeddings.tolist()

            # Check finish conditions
            for i, req in enumerate(batch.reqs):