From be98dc116ab62217d26dd067e89feb341616f283 Mon Sep 17 00:00:00 2001
From: ver217 <lhx0217@gmail.com>
Date: Wed, 24 Apr 2024 14:59:45 +0800
Subject: [PATCH] [test] fix llama test

---
 tests/kit/model_zoo/transformers/llama.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tests/kit/model_zoo/transformers/llama.py b/tests/kit/model_zoo/transformers/llama.py
index 58b5b0487a82..61fa560506c2 100644
--- a/tests/kit/model_zoo/transformers/llama.py
+++ b/tests/kit/model_zoo/transformers/llama.py
@@ -64,7 +64,6 @@ def data_gen_for_casual_lm():
         intermediate_size=64,
         num_attention_heads=4,
         max_position_embeddings=128,
-        num_labels=16,
     )
 
     if hasattr(config, "pad_token_id"):