Skip to content

Commit 3851e5f

Browse files
authored
[fix] update extra_rollout_info by input kwargs (#1268)
update extra_rollout_info by input
1 parent 8e2468f commit 3851e5f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

xtuner/v1/ray/config/worker.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -268,6 +268,12 @@ def __init__(self, **kwargs):
268268
else:
269269
kwargs["rollout_max_batch_size_per_instance"] = 128
270270

271+
default_extra_config = self.__class__.__fields__["extra_rollout_config"].default.copy()
272+
extra_config = kwargs.get("extra_rollout_config")
273+
if extra_config:
274+
default_extra_config.update(extra_config)
275+
kwargs["extra_rollout_config"] = default_extra_config
276+
271277
super().__init__(**kwargs)
272278
self.worker_log_dir.mkdir(parents=True, exist_ok=True)
273279

0 commit comments

Comments
 (0)