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

Add TogetherInferenceLLM #215

Merged
merged 8 commits into from
Jan 5, 2024
Merged

Add TogetherInferenceLLM #215

merged 8 commits into from
Jan 5, 2024

Conversation

alvarobartt
Copy link
Member

Description

This PR adds the TogetherInferenceLLM to support Together Inference Endpoints via their Python client. Also one example has been included under examples/ and the documentation has been updated accordingly.

Additionally, this PR also fixes a bug within the Argilla integration developed to unwrap the rows when using the LLMPool that was affecting the TextGenerationTask, which is now solved.

Closes #95

Copy link
Contributor

@plaguss plaguss left a comment

Choose a reason for hiding this comment

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

Awesome!

src/distilabel/llm/together.py Outdated Show resolved Hide resolved
Co-authored-by: Agus <agustin@argilla.io>
@alvarobartt alvarobartt merged commit 53ca00c into main Jan 5, 2024
4 checks passed
@alvarobartt alvarobartt deleted the together-inference branch January 5, 2024 13:12
gabrielmbmb added a commit that referenced this pull request Jan 5, 2024
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.

[FEATURE] Add support for Together Inference endpoints
3 participants