Fix post gemma merge (#31660)

* nit

* toctree issue

* protect gemma2 tests as well

* sdpa supported

docs/source/en/_toctree.yml

@@ -382,6 +382,8 @@
         title: Fuyu
       - local: model_doc/gemma
         title: Gemma
+      - local: model_doc/gemma2
+        title: Gemma2
       - local: model_doc/openai-gpt
         title: GPT
       - local: model_doc/gpt_neo

docs/source/en/perf_infer_gpu_one.md

@@ -43,6 +43,7 @@ FlashAttention-2 is currently supported for the following architectures:
 * [Dbrx](https://huggingface.co/docs/transformers/model_doc/dbrx#transformers.DbrxModel)
 * [DistilBert](https://huggingface.co/docs/transformers/model_doc/distilbert#transformers.DistilBertModel)
 * [Gemma](https://huggingface.co/docs/transformers/model_doc/gemma#transformers.GemmaModel)
+* [Gemma2](https://huggingface.co/docs/transformers/model_doc/gemma2#transformers.Gemma2Model)
 * [GPT2](https://huggingface.co/docs/transformers/model_doc/gpt2)
 * [GPTBigCode](https://huggingface.co/docs/transformers/model_doc/gpt_bigcode#transformers.GPTBigCodeModel)
 * [GPTNeo](https://huggingface.co/docs/transformers/model_doc/gpt_neo#transformers.GPTNeoModel)
@@ -202,6 +203,7 @@ For now, Transformers supports SDPA inference and training for the following arc
 * [Dpr](https://huggingface.co/docs/transformers/model_doc/dpr#transformers.DprReader)
 * [Falcon](https://huggingface.co/docs/transformers/model_doc/falcon#transformers.FalconModel)
 * [Gemma](https://huggingface.co/docs/transformers/model_doc/gemma#transformers.GemmaModel)
+* [Gemma2](https://huggingface.co/docs/transformers/model_doc/gemma2#transformers.Gemma2Model)
 * [GPT2](https://huggingface.co/docs/transformers/model_doc/gpt2)
 * [GPTBigCode](https://huggingface.co/docs/transformers/model_doc/gpt_bigcode#transformers.GPTBigCodeModel)
 * [GPTNeoX](https://huggingface.co/docs/transformers/model_doc/gpt_neox#transformers.GPTNeoXModel)

tests/models/gemma2/test_modeling_gemma2.py

@@ -41,11 +41,12 @@ if is_torch_available():
 
 
 class Gemma2ModelTester(GemmaModelTester):
-    config_class = Gemma2Config
-    model_class = Gemma2Model
-    for_causal_lm_class = Gemma2ForCausalLM
-    for_sequence_class = Gemma2ForSequenceClassification
-    for_token_class = Gemma2ForTokenClassification
+    if is_torch_available():
+        config_class = Gemma2Config
+        model_class = Gemma2Model
+        for_causal_lm_class = Gemma2ForCausalLM
+        for_sequence_class = Gemma2ForSequenceClassification
+        for_token_class = Gemma2ForTokenClassification
 
 
 @require_torch