transformers

pin `tensorflow_probability<0.22` in docker files

#34381

Merged