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

Update LLM and AsyncLLM to expose more functionality #90

Open
wants to merge 2 commits into
base: MLPerf_4.1
Choose a base branch
from

Conversation

gyulaz-htec
Copy link

Changed LLM class to provide a callback which can be used to insert additional requests to a running LLM evaluation

  • This can be used to better balance the load between multiple TP1 instances for offline inference.
    Updated AsyncLLMEngine with the following:
  • Added option to only send back first token(s) and the final result
  • Replaced execute_model_async with execute_model. According to rpd trace results this has lesser CPU overhead.

Zakor added 2 commits July 18, 2024 13:47
- Add option to only send back first token(s) and the final result
- Replaced execute_model_async with execute_model. According to rpd trace results this has lesser CPU overhead
@gyulaz-htec gyulaz-htec changed the title Update LLM and AsyncLLM related classes Update LLM and AsyncLLM to expose more functionality Jul 18, 2024
@shajrawi shajrawi requested a review from gshtras July 18, 2024 15:15
Copy link

This pull request has been automatically marked as stale because it has not had any activity within 90 days. It will be automatically closed if no further activity occurs within 30 days. Leave a comment if you feel this pull request should remain open. Thank you!

@github-actions github-actions bot added the stale label Oct 30, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant