Unverified Commit f52a43a8 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[ci][test] fix pp test failure (#7945)

parent e3580537
...@@ -126,7 +126,7 @@ class MultiprocessingGPUExecutor(DistributedGPUExecutor): ...@@ -126,7 +126,7 @@ class MultiprocessingGPUExecutor(DistributedGPUExecutor):
max_parallel_loading_workers) max_parallel_loading_workers)
def _check_executor_parameters(self): def _check_executor_parameters(self):
world_size = self.parallel_config.tensor_parallel_size world_size = self.parallel_config.world_size
tensor_parallel_size = self.parallel_config.tensor_parallel_size tensor_parallel_size = self.parallel_config.tensor_parallel_size
# Set CUDA_VISIBLE_DEVICES for the driver, inherited by workers # Set CUDA_VISIBLE_DEVICES for the driver, inherited by workers
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment