cog icon indicating copy to clipboard operation
cog copied to clipboard

Add Torch Dependency Resolution

Open 8W9aG opened this issue 1 year ago • 2 comments

  • Use our compatibility matrix to automatically resolve the torch version if the torch version is blank but adjacent dependencies are defined.

  • Make torchaudio capable of determining CUDA version compatibility like torchvision

  • Rewrite dependency versions to align with resolved versions

  • Remove version modifiers, this is now satisfied by adding in extra index URLs to the requirements

  • Fixes: https://github.com/replicate/cog/issues/1951

8W9aG avatar Sep 12 '24 21:09 8W9aG

Now that the tests run properly, that test_torch_version_resolution_with_torchvision test failure looks suspicious 👀

philandstuff avatar Sep 13 '24 10:09 philandstuff

Now that the tests run properly, that test_torch_version_resolution_with_torchvision test failure looks suspicious 👀

Looks like we are hitting the timeout on integration tests, I will bump to 20

8W9aG avatar Sep 13 '24 17:09 8W9aG