Skip to content

Commit

Permalink
fix path
Browse files Browse the repository at this point in the history
  • Loading branch information
fotstrt committed Nov 10, 2024
1 parent 8780f0d commit b3c5d6f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sailor/run_ft_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ def ConfigurationChange(self, request, context):
topology_list = list(request.topology)
if self.is_in_topo(topology_list):
print(f"Starting new process, node rank is {self.node_rank}")
start_cmd_base = f"python run_train_custom.py --config-file {self.script_args.config_file} --world-size {self.world_size} --master-ip {self.master_addr} --master-port {request.port}"
start_cmd_base = f"python /workspace/nanotron/sailor/run_train_custom.py --config-file {self.script_args.config_file} --world-size {self.world_size} --master-ip {self.master_addr} --master-port {request.port}"
for i in range(self.gpus_per_node):
print(f"Start for process {i}")
rank_i = self.node_rank*self.gpus_per_node + i
Expand Down

0 comments on commit b3c5d6f

Please sign in to comment.