xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 06:35:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
youkaichao 2024-08-28 01:27:07 -07:00 committed by GitHub
parent e3580537a4
commit f52a43a8b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/executor/multiproc_gpu_executor.py
View File

@ -126,7 +126,7 @@ class MultiprocessingGPUExecutor(DistributedGPUExecutor):
max_parallel_loading_workers)
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
# Set CUDA_VISIBLE_DEVICES for the driver, inherited by workers