Fix tfp dependency due to https://github.com/tensorflow/tensor2tensor/commit/d7d65f11faa40a44e9518c706c5accf8f83fb633
Fix tfp dependency due to https://github.com/tensorflow/tensor2tensor/commit/d7d65f11faa40a44e9518c706c5accf8f83fb633

PiperOrigin-RevId: 272800848

magenta/magenta,magenta/magenta