You can reproduce this issue by following the tutorial in https://www.tensorflow.org/text/tutorials/text_generation, forgive me the bad writing.

Good new, I update the libraries it seems like Apple developer has resolved this issue. There is the packages version details tensorflow        2.7.0      cpu_py39h2839aeb_0  conda-forge tensorflow-base      2.7.0      cpu_py39hbdae8f9_0  conda-forge tensorflow-datasets    4.3.0       pyhd8ed1ab_0  conda-forge tensorflow-deps      2.8.0             0  apple tensorflow-estimator   2.7.0      cpu_py39h91c69d6_0  conda-forge tensorflow-examples    eb925e460f761f5ed643d17f0c449e040ac2ac45-     pypi_0  pypi tensorflow-hub      0.12.0       pyhca92ed8_0  conda-forge tensorflow-macos     2.8.0          pypi_0  pypi tensorflow-metadata    1.2.0       pyhd8ed1ab_0  conda-forge tensorflow-metal     0.4.0          pypi_0  pypi