diff --git a/tests/kit/model_zoo/transformers/gpt.py b/tests/kit/model_zoo/transformers/gpt.py index 67b648382..44d859e48 100644 --- a/tests/kit/model_zoo/transformers/gpt.py +++ b/tests/kit/model_zoo/transformers/gpt.py @@ -113,7 +113,7 @@ config = transformers.GPT2Config( problem_type="single_label_classification", pad_token_id=1022, tie_word_embeddings=True, - attn_implementation="eager" + attn_implementation="eager", ) config_for_token_classification = copy.deepcopy(config)