search

NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
https://nvidia-merlin.github.io/Transformers4Rec/main
Apache License 2.0
1.08k stars 142 forks source link

Add Transformers Torch Extras to install requirements #699

Closed oliverholworthy closed 1 year ago

oliverholworthy commented 1 year ago

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.

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/Transformers4Rec/review/pr-699