We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d7a03bf commit d3c40b9Copy full SHA for d3c40b9
tests/kit/model_zoo/transformers/gpt.py
@@ -113,7 +113,7 @@ def date_gen_for_double_heads():
113
problem_type="single_label_classification",
114
pad_token_id=1022,
115
tie_word_embeddings=True,
116
- attn_implementation="eager"
+ attn_implementation="eager",
117
)
118
119
config_for_token_classification = copy.deepcopy(config)
0 commit comments