Path Lines of Code optimum/tpu/modeling_gemma.py 877 optimum/tpu/modeling_llama.py 1124 optimum/tpu/modeling_mistral.py 1106 optimum/tpu/static_cache_xla.py 23 text-generation-inference/server/text_generation_server/cli.py 78 text-generation-inference/server/text_generation_server/generator.py 681 text-generation-inference/server/text_generation_server/jetstream_pt_support/generator.py 439 text-generation-inference/server/text_generation_server/jetstream_pt_support/models/llama_model_exportable_hf.py 102