release

flybird11111 · flybird11111 · commit 45779680bfb1 · 2025-05-27T15:10:19.000+08:00
diff --git a/tests/kit/model_zoo/transformers/gpt.py b/tests/kit/model_zoo/transformers/gpt.py
@@ -113,7 +113,7 @@ def date_gen_for_double_heads():
     problem_type="single_label_classification",
     pad_token_id=1022,
     tie_word_embeddings=True,
-    attn_implementation="flash_attention_2",
+    attn_implementation="eager",
 )
 
 config_for_token_classification = copy.deepcopy(config)

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ def date_gen_for_double_heads():`
`113`	`113`	`problem_type="single_label_classification",`
`114`	`114`	`pad_token_id=1022,`
`115`	`115`	`tie_word_embeddings=True,`
`116`		`- attn_implementation="flash_attention_2",`
	`116`	`+ attn_implementation="eager",`
`117`	`117`	`)`
`118`	`118`
`119`	`119`	`config_for_token_classification = copy.deepcopy(config)`