-
Notifications
You must be signed in to change notification settings - Fork 36
New issue
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
2x slowdown using TP #117
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks for this very interesting lib @BlackSamorez! I just tried running your kaggle nb locally, using 2 x A6000s. I used the 'meta-llama/Llama-2-7b-hf' model. The call to
model.generate
withmax_length=200
takes 12 seconds when usingtensor_parallel
with the 2 GPUs.However, if I remove
tensor_parallel
and instead just use a single GPU, generation is over twice as fast, taking 5 seconds.Is this slowdown expected, or am I doing something wrong?
The text was updated successfully, but these errors were encountered: