We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8e2468f commit 3851e5fCopy full SHA for 3851e5f
xtuner/v1/ray/config/worker.py
@@ -268,6 +268,12 @@ def __init__(self, **kwargs):
268
else:
269
kwargs["rollout_max_batch_size_per_instance"] = 128
270
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
+
277
super().__init__(**kwargs)
278
self.worker_log_dir.mkdir(parents=True, exist_ok=True)
279
0 commit comments