File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change 2727 RayWorkerGroup ,
2828)
2929from verl .single_controller .ray .base import create_colocated_worker_cls
30+ from verl .utils .config import omega_conf_to_dataclass
3031from verl .utils .device import get_device_name
3132from verl .utils .tokenizer import hf_tokenizer
33+ from verl .workers .config import CheckpointEngineConfig
3234from verl .workers .engine_workers import ActorRolloutRefWorker
3335
3436
@@ -97,8 +99,11 @@ def test_server_adapter_colocated_weight_update(init_config):
9799 )
98100
99101 # 2. create CheckpointEngineManager
102+ checkpoint_engine_config : CheckpointEngineConfig = omega_conf_to_dataclass (
103+ init_config .actor_rollout_ref .rollout .checkpoint_engine
104+ )
100105 checkpoint_manager = CheckpointEngineManager (
101- backend = init_config . actor_rollout_ref . rollout . checkpoint_engine . backend ,
106+ config = checkpoint_engine_config ,
102107 trainer = actor_rollout_wg ,
103108 replicas = agent_loop_manager .rollout_replicas ,
104109 )
You can’t perform that action at this time.
0 commit comments