Skip to content
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

fix: Fix Triton server model loading error #584

Merged
merged 1 commit into from
Jul 15, 2024

Conversation

ratnopamc
Copy link
Collaborator

What does this PR do?

One of the recent PRs introduced an error(see below) causing the model loading to fail when the triton server gets deployed.

Error Log:

I0715 04:06:56.609115 1 server.cc:674] 
+-----------+---------+---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
| Model     | Version | Status                                                                                                                                                                                                        |
+-----------+---------+---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
| llama2    | 1       | READY                                                                                                                                                                                                         |
| llama3    | 1       | UNAVAILABLE: Invalid argument: instance group llama3_0 of model llama3 specifies invalid or unsupported gpu id 2. GPUs with at least the minimum required CUDA compute compatibility of 6.000000 are: 0       |
| mistral7b | 1       | UNAVAILABLE: Invalid argument: instance group mistral7b_0 of model mistral7b specifies invalid or unsupported gpu id 1. GPUs with at least the minimum required CUDA compute compatibility of 6.000000 are: 0 |
+-----------+---------

This PR fixes the above issue.

🛑 Please open an issue first to discuss any significant work and flesh out details/direction - we would hate for your time to be wasted.
Consult the CONTRIBUTING guide for submitting pull-requests.

Motivation

More

  • Yes, I have tested the PR using my local account setup (Provide any test evidence report under Additional Notes)
  • Mandatory for new blueprints. Yes, I have added a example to support my blueprint PR
  • Mandatory for new blueprints. Yes, I have updated the website/docs or website/blog section for this feature
  • Yes, I ran pre-commit run -a with this PR. Link for installing pre-commit locally

For Moderators

  • E2E Test successfully complete before merge?

Additional Notes

@ratnopamc ratnopamc requested a review from vara-bonthu July 15, 2024 04:26
Copy link
Collaborator

@vara-bonthu vara-bonthu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the fix @ratnopamc !!

@vara-bonthu vara-bonthu merged commit be5e370 into awslabs:main Jul 15, 2024
37 of 38 checks passed
ovaleanu pushed a commit to ovaleanu/data-on-eks that referenced this pull request Aug 10, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants