Remove Yi model definition, please use LlamaForCausalLM instead (#2854)

Co-authored-by: Roy <jasonailu87@gmail.com>
2024-02-13 14:22:22 -08:00
parent a463c333dd
commit 317b29de0f
5 changed files with 2 additions and 402 deletions
--- a/vllm/transformers_utils/config.py
+++ b/vllm/transformers_utils/config.py
@@ -12,7 +12,6 @@ _CONFIG_REGISTRY = {
    "qwen": QWenConfig,
    "RefinedWeb": RWConfig,  # For tiiuae/falcon-40b(-instruct)
    "RefinedWebModel": RWConfig,  # For tiiuae/falcon-7b(-instruct)
-    "yi": YiConfig,
 }