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

vLLM Hugging Face exporter #11124

Merged
merged 43 commits into from
Nov 4, 2024
Merged

Conversation

oyilmaz-nvidia
Copy link
Collaborator

What does this PR do ?

Adds an exporter for the HF models to vLLM and deploys it with PyTriton.

oyilmaz-nvidia and others added 30 commits June 13, 2024 10:12
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
akoumpa
akoumpa previously approved these changes Nov 1, 2024
Copy link
Member

@akoumpa akoumpa left a comment

Choose a reason for hiding this comment

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

LGTM, thanks.

@akoumpa akoumpa enabled auto-merge (squash) November 1, 2024 17:29
@akoumpa akoumpa merged commit 65415ed into NVIDIA:main Nov 4, 2024
159 of 160 checks passed
hainan-xv pushed a commit to hainan-xv/NeMo that referenced this pull request Nov 5, 2024
* fix minor import bug

Signed-off-by: Onur Yilmaz <[email protected]>

* Add save pretrained to HF model

Signed-off-by: Onur Yilmaz <[email protected]>

* vllm hf exporter added

Signed-off-by: Onur Yilmaz <[email protected]>

* pytriton function added

Signed-off-by: Onur Yilmaz <[email protected]>

* update test

Signed-off-by: Onur Yilmaz <[email protected]>

* Example fixed

Signed-off-by: Onur Yilmaz <[email protected]>

* Apply isort and black reformatting

Signed-off-by: oyilmaz-nvidia <[email protected]>

* Remove unused import

---------

Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: oyilmaz-nvidia <[email protected]>
Co-authored-by: oyilmaz-nvidia <[email protected]>
Signed-off-by: Hainan Xu <[email protected]>
lilyw97 pushed a commit to lilyw97/NeMo that referenced this pull request Nov 13, 2024
* fix minor import bug

Signed-off-by: Onur Yilmaz <[email protected]>

* Add save pretrained to HF model

Signed-off-by: Onur Yilmaz <[email protected]>

* vllm hf exporter added

Signed-off-by: Onur Yilmaz <[email protected]>

* pytriton function added

Signed-off-by: Onur Yilmaz <[email protected]>

* update test

Signed-off-by: Onur Yilmaz <[email protected]>

* Example fixed

Signed-off-by: Onur Yilmaz <[email protected]>

* Apply isort and black reformatting

Signed-off-by: oyilmaz-nvidia <[email protected]>

* Remove unused import

---------

Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: oyilmaz-nvidia <[email protected]>
Co-authored-by: oyilmaz-nvidia <[email protected]>
HuiyingLi pushed a commit to HuiyingLi/NeMo that referenced this pull request Nov 15, 2024
* fix minor import bug

Signed-off-by: Onur Yilmaz <[email protected]>

* Add save pretrained to HF model

Signed-off-by: Onur Yilmaz <[email protected]>

* vllm hf exporter added

Signed-off-by: Onur Yilmaz <[email protected]>

* pytriton function added

Signed-off-by: Onur Yilmaz <[email protected]>

* update test

Signed-off-by: Onur Yilmaz <[email protected]>

* Example fixed

Signed-off-by: Onur Yilmaz <[email protected]>

* Apply isort and black reformatting

Signed-off-by: oyilmaz-nvidia <[email protected]>

* Remove unused import

---------

Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: oyilmaz-nvidia <[email protected]>
Co-authored-by: oyilmaz-nvidia <[email protected]>
yashaswikarnati pushed a commit that referenced this pull request Nov 21, 2024
* fix minor import bug

Signed-off-by: Onur Yilmaz <[email protected]>

* Add save pretrained to HF model

Signed-off-by: Onur Yilmaz <[email protected]>

* vllm hf exporter added

Signed-off-by: Onur Yilmaz <[email protected]>

* pytriton function added

Signed-off-by: Onur Yilmaz <[email protected]>

* update test

Signed-off-by: Onur Yilmaz <[email protected]>

* Example fixed

Signed-off-by: Onur Yilmaz <[email protected]>

* Apply isort and black reformatting

Signed-off-by: oyilmaz-nvidia <[email protected]>

* Remove unused import

---------

Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: oyilmaz-nvidia <[email protected]>
Co-authored-by: oyilmaz-nvidia <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants