diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py index 798d3a6..b6118a5 100644 --- a/tests/test_hf_model.py +++ b/tests/test_hf_model.py @@ -203,7 +203,7 @@ class TestMMModel: do_sample=False) print(response) assert len(response) != 0 - assert ' panda ' in response + assert ' panda' in response query = ' 请根据图片写一篇作文:我最喜欢的小动物。' \ + '要求:选准角度,确定立意,明确文体,自拟标题。'