diff --git a/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_38b.py b/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_38b.py index 971ec947..98713696 100644 --- a/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_38b.py +++ b/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_38b.py @@ -1,8 +1,8 @@ -from opencompass.models import TurboMindModel +from opencompass.models import TurboMindModelwithChatTemplate models = [ dict( - type=TurboMindModel, + type=TurboMindModelwithChatTemplate, abbr='internvl2_5-38b-turbomind', path='OpenGVLab/InternVL2_5-38B', engine_config=dict(session_len=8192, max_batch_size=8, tp=4), diff --git a/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_8b.py b/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_8b.py index 68e3b5be..3541249c 100644 --- a/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_8b.py +++ b/opencompass/configs/models/intervl/lmdeploy_intervl_2_5_8b.py @@ -1,8 +1,8 @@ -from opencompass.models import TurboMindModel +from opencompass.models import TurboMindModelwithChatTemplate models = [ dict( - type=TurboMindModel, + type=TurboMindModelwithChatTemplate, abbr='internvl2_5-8b-turbomind', path='OpenGVLab/InternVL2_5-8B', engine_config=dict(session_len=8192, max_batch_size=16, tp=1),