From abf6016f2ee69aa50e77ab19e878bbca99af2b8e Mon Sep 17 00:00:00 2001
From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com>
Date: Fri, 1 Mar 2024 09:43:17 +0800
Subject: [PATCH] add xcomposer 4bit and internlm2-chat-20b test case

---
 tests/test_hf_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = '<ImageHere> <ImageHere>请根据图片写一篇作文：我最喜欢的小动物。' \
             + '要求：选准角度，确定立意，明确文体，自拟标题。'