v0.1: Initial Release
This executor takes Jina's TransformerTorchEncoder, and updates it to:
- support CUDA 11.3
- provide arm64 CPU-only images so the container can run without very slow emulation on M1 macs.
Using from Jina Hub
Flow().add(uses='jinahub+docker://TransformerTorchEncoderCU113/v0.1')
# or
Flow().add(uses='jinahub+docker://TransformerTorchEncoderCU113/v0.1-gpu')
Using from GHCR
# on amd64 or arm64 systems:
Flow().add(uses='docker://ghcr.io/freddyheppell/transformer-torch-encoder:v0.1')
# on amd64 systems:
Flow().add(uses='docker://ghcr.io/freddyheppell/transformer-torch-encoder:v0.1-gpu')