Skip to content

Commit bde1e90

Browse files
[misc] add sglang support for hdfs file load (volcengine#1060)
1 parent 26e0014 commit bde1e90

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

verl/workers/fsdp_workers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -353,7 +353,8 @@ def _build_rollout(self, trust_remote_code=False):
353353
# check: https://github.com/sgl-project/sglang/blob/00f42707eaddfc2c0528e5b1e0094025c640b7a0/python/sglang/srt/layers/quantization/fp8_utils.py#L76
354354
from verl.workers.sharding_manager.fsdp_sglang import FSDPSGLangShardingManager
355355
log_gpu_memory_usage(f'Before building {rollout_name} rollout', logger=None)
356-
rollout = SGLangRollout(actor_module=self.config.model.path,
356+
local_path = copy_to_local(self.config.model.path)
357+
rollout = SGLangRollout(actor_module=local_path,
357358
config=self.config.rollout,
358359
tokenizer=self.tokenizer,
359360
model_hf_config=self.actor_model_config)

0 commit comments

Comments
 (0)