fix gemma bug #2727

Job	Run time
build-and-test-cpp-aarch64 (openblas)	1m 42s
build-python-wheels (ubuntu-20.04, auto64)	1m 40s
build-and-push-docker-images	1m 41s
build-python-wheels (macos-11, auto64)	1m 39s
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	1m 40s
check-python-style	11s
build-python-wheels (windows-2019, auto64)	1m 30s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	1m 36s
build-python-wheels (ubuntu-20.04, aarch64)	1m 41s
build-and-test-cpp-x86_64 (macos-11, mkl)	1m 39s
build-python-wheels (macos-11, arm64)	1m 40s
build-and-test-cpp-x86_64 (macos-11, dnnl)	1m 40s
test-python-wheels	1s
build-and-deploy-docs	1s
publish-python-wheels-on-pypi	1s
	18m 22s

Provide feedback