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 missing google-cloud-storage dependency in PyTorch DLC #124

Merged
merged 1 commit into from
Nov 29, 2024

Conversation

alvarobartt
Copy link
Member

Description

This PR is a follow up of #120 as reported by @xuyifann due to a missing dependency for google-cloud-storage within the Hugging Face PyTorch DLC for Inference on both CPU and GPU for the transformers version 4.41.0; as when patching the DLC in #120, that dependency was missing, but was catched during the testing after building the DLC.

This PR then adds the missing google-cloud-storage dependency within the Dockerfile, to patch that exact DLC, whilst the rest work fine, since those dependencies are installed as part of the google extra of huggingface-inference-toolkit, not available within the pinned huggingface-inference-toolkit version installed in this DLC in particular.

@alvarobartt alvarobartt added pytorch Pytorch related Issues fix labels Nov 28, 2024
@alvarobartt alvarobartt requested a review from xuyifann November 28, 2024 10:34
@alvarobartt alvarobartt self-assigned this Nov 28, 2024
@alvarobartt alvarobartt changed the title Fix PyTorch DLC for Inference (transformers 4.41.0) Add missing google-cloud-storage dependency in PyTorch DLC Nov 28, 2024
@alvarobartt alvarobartt merged commit 5baddf5 into main Nov 29, 2024
1 check passed
@alvarobartt alvarobartt deleted the fix-pytorch-inference-4.41 branch November 29, 2024 08:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
fix pytorch Pytorch related Issues
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants