We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
I want to set PIPELINE_PARALLEL_SIZE instead of TENSOR_PARALLEL_SIZE for multiple GPUs.
PIPELINE_PARALLEL_SIZE
TENSOR_PARALLEL_SIZE
But tensor_parallel_size is always set to number of GPU.
tensor_parallel_size
It's needed to check whether the env was set or not in engine_args.py:
engine_args.py
num_gpus = device_count() if num_gpus > 1: args["tensor_parallel_size"] = num_gpus
The text was updated successfully, but these errors were encountered:
No branches or pull requests
I want to set
PIPELINE_PARALLEL_SIZE
instead ofTENSOR_PARALLEL_SIZE
for multiple GPUs.But
tensor_parallel_size
is always set to number of GPU.It's needed to check whether the env was set or not in
engine_args.py
:The text was updated successfully, but these errors were encountered: