From d84af04d38eaf9564de127ab0000dce1a2761882 Mon Sep 17 00:00:00 2001 From: wgzintel Date: Thu, 25 Apr 2024 15:26:33 +0800 Subject: [PATCH] enable phi2 models --- llm_bench/python/utils/config_class.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llm_bench/python/utils/config_class.py b/llm_bench/python/utils/config_class.py index 04334e24bf..2f11cc388e 100644 --- a/llm_bench/python/utils/config_class.py +++ b/llm_bench/python/utils/config_class.py @@ -95,6 +95,7 @@ 'mixtral', 'yi-', 'phi-', + 'phi2-', 'minicpm', 'gemma', "deci",