-
Notifications
You must be signed in to change notification settings - Fork 6k
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
[data/llm/docs] Initial draft of user guide for Data LLM APIs #50674
Conversation
ci/docker/llm.build.Dockerfile
Outdated
# FIXME(comaniac) | ||
pip uninstall -y torch | ||
pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121 | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
remove
Configure vLLM for LLM inference | ||
-------------------------------- | ||
|
||
Use the `VLLMProcessorConfig` to configure the vLLM engine. | ||
|
||
.. testcode:: | ||
|
||
from ray.data.llm import VLLMProcessorConfig | ||
|
||
processor_config = VLLMProcessorConfig( | ||
model="meta-llama/Llama-3.1-8B-Instruct", | ||
... | ||
) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Question - What other tips should we add here about this
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Look at vllm docs for examples on setting good args too
|
||
from ray.data.llm import VLLMProcessorConfig | ||
|
||
processor_config = VLLMProcessorConfig( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this is small v
- vLLMProcessorConfig
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed
@@ -12,6 +12,9 @@ Offline batch inference is a process for generating model predictions on a fixed | |||
:width: 650px | |||
:align: center | |||
|
|||
.. note:: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Add an example for a custom evaluation pipeline, reading from huggingface etc
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I will add this later
Signed-off-by: Richard Liaw <[email protected]>
c144bf9
to
5568401
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed the examples. Otherwise LGTM
Co-authored-by: Cody Yu <[email protected]> Signed-off-by: Richard Liaw <[email protected]>
Signed-off-by: Richard Liaw <[email protected]>
Co-authored-by: Cody Yu <[email protected]> Signed-off-by: Richard Liaw <[email protected]>
Signed-off-by: Richard Liaw <[email protected]>
Signed-off-by: Richard Liaw <[email protected]>
…oject#50674) ## Why are these changes needed? Adds user guide and link-ins for Ray Data documentation. This is part of the ray-project#50639 thread of work. This is based on ray-project#50494 cc @comaniac @gvspraveen @kouroshHakha ## Related issue number <!-- For example: "Closes ray-project#1234" --> ## Checks - [ ] I've signed off every commit(by using the -s flag, i.e., `git commit -s`) in this PR. - [ ] I've run `scripts/format.sh` to lint the changes in this PR. - [ ] I've included any doc changes needed for https://docs.ray.io/en/master/. - [ ] I've added any new APIs to the API Reference. For example, if I added a method in Tune, I've added it in `doc/source/tune/api/` under the corresponding `.rst` file. - [ ] I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/ - Testing Strategy - [ ] Unit tests - [ ] Release tests - [ ] This PR is not tested :( --------- Signed-off-by: Richard Liaw <[email protected]> Co-authored-by: Cody Yu <[email protected]> Signed-off-by: 400Ping <[email protected]>
Why are these changes needed?
Adds user guide and link-ins for Ray Data documentation.
This is part of the #50639 thread of work.
This is based on #50494
cc @comaniac @gvspraveen @kouroshHakha
Related issue number
Checks
git commit -s
) in this PR.scripts/format.sh
to lint the changes in this PR.method in Tune, I've added it in
doc/source/tune/api/
under thecorresponding
.rst
file.