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 Ray for PyTorch DLCs for Inference / Training #129

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

alvarobartt
Copy link
Member

@alvarobartt alvarobartt commented Dec 5, 2024

Description

This PR is tentative and just for exploring whether it makes sense to include Ray (and the required extra dependencies to make it work) as a core dependency of the Hugging Face PyTorch DLC for Inference / Training.

The size is about ~700Mb more for the PyTorch DLC for Inference and ~400MB more for the PyTorch DLC for Training, compared to their base existing DLC counterparts.

cc @inardini for visibility and @philschmid to get your opinion on whether it should be a core dependency of the DLC or kept separate

@alvarobartt alvarobartt self-assigned this Dec 5, 2024
@alvarobartt alvarobartt changed the title Add Ray and python-json-logger for Ray DLC Add Ray for PyTorch DLCs for Inference / Training Dec 6, 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.

1 participant