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(