jina-ai / executors

internal-only
Apache License 2.0
31 stars 12 forks source link

feat(gpu): split video torch encoder cpu gpu #180

Closed jacobowitz closed 3 years ago

jacobowitz commented 3 years ago

This PR mainly splits the docker images of VideoTorchEncoder into separate gpu and cpu images. This is done by separating both Docker files and requirements files. Also I fixed Jina usage by replacing the Docker parent image and removing hard coded jina version requirements.

Due to linting I also had to make a couple of more changes so the change is unnecessarily big, sorry for that.

This belongs to https://github.com/jina-ai/executors/issues/169 - I want to do one PR per executor