add comments

PaddlePaddle · Aug 31, 2023 · 03db007 · 03db007
1 parent 29f6725
commit 03db007
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/transformers/chatglm_v2/test_modeling.py b/tests/transformers/chatglm_v2/test_modeling.py
@@ -221,9 +221,11 @@ def test_model_attention_mask(self):
         config_and_inputs = self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_model_attention_mask(*config_and_inputs)
 
+    # chatglm_v2 cannot use beam search temporarily
     def test_beam_search_generate(self):
         pass
 
+    # chatglm_v2 cannot use group beam search temporarily
     def test_group_beam_search_generate(self):
         pass