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

feat: Sagemaker integration: SagemakerGenerator #276

Merged
merged 18 commits into from
Jan 30, 2024
Merged

Conversation

ZanSara
Copy link
Contributor

@ZanSara ZanSara commented Jan 26, 2024

Related Issues

Proposed Changes:

  • Add SagemakerGenerator with a very basic set of features (no streaming, no chat, only text inference)
  • Add tests

How did you test it?

Local unit and integration tests, unit tests in CI.

NOTE: integration tests won't run in the CI for now - keeping up the endpoint for these tests is very expensive.

Notes for the reviewer

  • Tested on Falcon models
  • Tested on Llama2 models

As keeping Sagemaker endpoints up for testing is very expensive, integration tests are not going to run in CI.

@ZanSara ZanSara changed the title feat: Sagemaker integration feat: Sagemaker integration: SagemakerGenerator Jan 26, 2024
@ZanSara ZanSara marked this pull request as ready for review January 29, 2024 10:44
@ZanSara ZanSara requested a review from a team as a code owner January 29, 2024 10:44
@ZanSara ZanSara requested review from vblagoje and removed request for a team January 29, 2024 10:44
Copy link
Member

@vblagoje vblagoje left a comment

Choose a reason for hiding this comment

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

Minor changes and updates requested :-)

@ZanSara ZanSara requested a review from vblagoje January 29, 2024 13:59
Copy link
Member

@vblagoje vblagoje left a comment

Choose a reason for hiding this comment

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

Looks much better now, let's 🚢

@ZanSara ZanSara merged commit 01f08b9 into main Jan 30, 2024
9 checks passed
@ZanSara ZanSara deleted the sagemaker-integration branch January 30, 2024 08:56
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.

Implement SagemakerGenerator
2 participants