Update STS HMC fitting docstring to demonstrate TF2/eager mode use.
Update STS HMC fitting docstring to demonstrate TF2/eager mode use.

PiperOrigin-RevId: 291838129

tensorflow/probability,tensorflow/probability