Skip to content

Add Transformers Torch Extras to install requirements #699

Merged
oliverholworthy merged 3 commits intoNVIDIA-Merlin:mainfrom
oliverholworthy:transformers-torch-dep
May 12, 2023
Merged

Add Transformers Torch Extras to install requirements #699
oliverholworthy merged 3 commits intoNVIDIA-Merlin:mainfrom
oliverholworthy:transformers-torch-dep

Conversation

@oliverholworthy
Copy link
Copy Markdown
Contributor

Adds torch extra from transformers package to base requirements.

This results in moving the torch dependency from an optional depdendency to an install depdendency. Since Transformers4Rec is only implemented for Torch and won't work without it, this seems ok and simplifies the pip install for the package.

We'll also pickup any depdendencies added to the torch extras in transformers like the recently added accelerate dependency.

@oliverholworthy oliverholworthy added the chore Maintenance for the repository label May 12, 2023
@oliverholworthy oliverholworthy added this to the Merlin 23.05 milestone May 12, 2023
@oliverholworthy oliverholworthy self-assigned this May 12, 2023
@oliverholworthy oliverholworthy force-pushed the transformers-torch-dep branch from a25a0e0 to f981d8e Compare May 12, 2023 13:40
@github-actions
Copy link
Copy Markdown

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

chore Maintenance for the repository

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants