Organize dependencies
-
Declare dependencies for each model as an
extra. This means that users would only have to download dependencies for the model they want to use by including its name in the installation command. -
Add
pytorchindex as a secondary source for poetry. Thetorchpackage on PyPI is the CUDA version that installs proprietary NVIDIA packages, as pointed out in T325349. Since we don't need CUDA, the source points to the cpu index. Also, becausetorchalready exists on PyPI, this extra index should not make the package more susceptible to dependency confusion attacks.