add xcomposer 4bit and internlm2-chat-20b test case

2024-03-01 09:43:17 +08:00 · 2024-03-01 09:43:17 +08:00 · abf6016f2e
parent 92b15b180d
commit abf6016f2e
1 changed files with 1 additions and 1 deletions
--- 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 = '<ImageHere> <ImageHere>请根据图片写一篇作文：我最喜欢的小动物。' \
            + '要求：选准角度，确定立意，明确文体，自拟标题。'