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 #91

Open
wants to merge 2 commits into
base: main
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 a lesser CPU (few ms) overhead.

Zakor added 2 commits July 18, 2024 17:46
- 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
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.

1 participant