SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix post gemma merge (#31660)

Browse Source 
* nit

* toctree issue

* protect gemma2 tests as well

* sdpa supported
This commit is contained in:
Arthur
2024-06-27 17:51:42 +02:00
committed by GitHub
parent 727eea4ab0
commit 75a6319864
3 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/models/gemma2/test_modeling_gemma2.py
View File

@@ -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