From 5a0ad9726b722076864d0479e33cf7ba2623271b Mon Sep 17 00:00:00 2001 From: Jonas Prellberg Date: Wed, 19 Nov 2025 14:28:29 +0100 Subject: [PATCH] [megatron] fix: pass trust_remote_code to get_generation_config --- verl/workers/megatron_workers.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/verl/workers/megatron_workers.py b/verl/workers/megatron_workers.py index cfd7b24f136..91e00e7adae 100644 --- a/verl/workers/megatron_workers.py +++ b/verl/workers/megatron_workers.py @@ -314,7 +314,10 @@ def _build_model_optimizer( self.config.model.get("trust_remote_code", False), self.config.actor.megatron.use_mbridge, ) - self.generation_config = get_generation_config(self.local_path) + self.generation_config = get_generation_config( + self.local_path, + self.config.model.get("trust_remote_code", False), + ) if self._is_actor or self._is_rollout: wrap_config = McoreModuleWrapperConfig(