Oh got it, thanks! Changed it to tao-toolkit-lm:v3.22.05-py3 and it works now
tao-toolkit-lm:v3.22.05-py3