Skip to content

[docker] update vllm wheel in neuronx container#2688

Merged
sindhuvahinis merged 1 commit intomasterfrom
vllmwheel
Jan 27, 2025
Merged

[docker] update vllm wheel in neuronx container#2688
sindhuvahinis merged 1 commit intomasterfrom
vllmwheel

Conversation

@sindhuvahinis
Copy link
Contributor

@sindhuvahinis sindhuvahinis commented Jan 27, 2025

Description

Brief description of what this PR is about

This PR updates the wheel to neuron vllm. https://github.com/aws-neuron/upstreaming-to-vllm. Built a wheel using internal wheel with vllm forked inside.

Type of change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • New feature (non-breaking change which adds functionality)
  • This change requires a documentation update

Checklist:

  • Please add the link of Integration Tests Executor run with related tests.
  • Have you manually built the docker image and verify the change?
  • Have you run related tests? Check how to set up the test environment here; One example would be pytest tests.py -k "TestCorrectnessLmiDist" -m "lmi_dist"
  • Have you added tests that prove your fix is effective or that this feature works?
  • Has code been commented, particularly in hard-to-understand areas?
  • Have you made corresponding changes to the documentation?

Feature/Issue validation/testing

Please describe the Unit or Integration tests that you ran to verify your changes and relevant result summary. Provide instructions so it can be reproduced.
Please also list any relevant details for your test configuration.

  • Test A
    Logs for Test A

  • Test B
    Logs for Test B

@sindhuvahinis sindhuvahinis requested review from a team and zachgk as code owners January 27, 2025 19:57
@sindhuvahinis sindhuvahinis merged commit 293da5a into master Jan 27, 2025
10 of 11 checks passed
@sindhuvahinis sindhuvahinis deleted the vllmwheel branch January 27, 2025 21:21
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.

3 participants