cog
cog copied to clipboard
Support Torch 2.4.0 base images
Currently seeing Cog doesn't know what CUDA version is compatible with torch==2.4.0 when working with images using torch 2.4.0, we need to add a base image to support this.
I agree with it
We now support up to torch 2.4.1 (coming in the next version)