Missing distillation cfg in ray_trainer

JacobHelwig · JacobHelwig · commit a087ed2f6705 · 2026-02-26T11:03:56.000-06:00
diff --git a/verl/trainer/ppo/ray_trainer.py b/verl/trainer/ppo/ray_trainer.py
@@ -698,7 +698,7 @@ def init_workers(self):
             actor_rollout_cls = RayClassWithInitArgs(
                 cls=self.role_worker_mapping[actor_role],
                 config=self.config.actor_rollout_ref,
-                distillation_config=self.config.distillation,
+                distillation_config=self.config.get("distillation"),
                 role=str(actor_role),
             )
             self.resource_pool_to_cls[actor_rollout_resource_pool][str(actor_role)] = actor_rollout_cls

Original file line number	Diff line number	Diff line change
`@@ -698,7 +698,7 @@ def init_workers(self):`
`698`	`698`	`actor_rollout_cls = RayClassWithInitArgs(`
`699`	`699`	`cls=self.role_worker_mapping[actor_role],`
`700`	`700`	`config=self.config.actor_rollout_ref,`
`701`		`- distillation_config=self.config.distillation,`
	`701`	`+ distillation_config=self.config.get("distillation"),`
`702`	`702`	`role=str(actor_role),`
`703`	`703`	`)`
`704`	`704`	`self.resource_pool_to_cls[actor_rollout_resource_pool][str(actor_role)] = actor_rollout_cls`