File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change 1313from typing import Dict , List
1414
1515import torch
16- from forge .controller .service import ServiceConfig
17- from forge .controller .spawn import spawn_service
1816
1917from forge .data .sharding import VLLMSharding
2018from forge .interfaces import Policy as PolicyInterface
2119from monarch .actor import Actor , current_rank , endpoint , proc_mesh
22- from omegaconf import DictConfig
2320from torchstore import MultiProcessStore
2421from torchstore ._state_dict_utils import DELIM
2522
@@ -337,7 +334,6 @@ async def setup(self, store: MultiProcessStore = None):
337334 self .torchstore = store
338335 # TODO: remove ["gpus"] when monarch implements a flat rank
339336 self .rank = current_rank ()["gpus" ]
340- # print(f" RAnk: {self.rank}")
341337 self .worker = self .setup_worker ()
342338
343339 @endpoint
You can’t perform that action at this time.
0 commit comments