Update keras_hub/src/samplers/top_p_sampler.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: AlejandroPG06

keras_hub/src/samplers/top_p_sampler.py

@@ -81,7 +81,7 @@ def get_next_token(self, probabilities):
         sorted_next_token = random.categorical(
             # tf does not support half precision multinomial sampling, so make
             # sure we have full precision here.
-            ops.cast(ops.log(probabilities), "float32"),
+            ops.log(ops.cast(probabilities, "float32")),
             1,
             seed=self.seed_generator,
             dtype="int32",