fix rl bugs (#3654)

This commit is contained in:
YuanRisheng
2025-08-28 11:09:34 +08:00
committed by GitHub
parent e81046fdad
commit ad9b95e6dd

View File

@@ -1158,10 +1158,6 @@ class FDConfig:
""" """
calculate some parameters calculate some parameters
""" """
assert (
self.device_ids.split(",").__len__() == self.worker_num_per_node
), f"invalid CUDA_VISIBLE_DEVICES, should be equal to {self.worker_num_per_node}"
self.local_device_ids = self.device_ids.split(",")[: self.parallel_config.tensor_parallel_size] self.local_device_ids = self.device_ids.split(",")[: self.parallel_config.tensor_parallel_size]
if self.parallel_config.tensor_parallel_size <= self.worker_num_per_node: if self.parallel_config.tensor_parallel_size <= self.worker_num_per_node: