From 69031fd1557ce3ef7017c2e9466e70245d458bd7 Mon Sep 17 00:00:00 2001 From: xiaoji <44150358+YeFD@users.noreply.github.com> Date: Mon, 22 Apr 2024 00:57:08 +0800 Subject: [PATCH] [Doc]: Update the doc of adding new models (#4236) --- docs/source/models/adding_model.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/models/adding_model.rst b/docs/source/models/adding_model.rst index a82c2cef10e83..cbc8099e6f70f 100644 --- a/docs/source/models/adding_model.rst +++ b/docs/source/models/adding_model.rst @@ -95,7 +95,7 @@ This method should load the weights from the HuggingFace's checkpoint file and a 5. Register your model ---------------------- -Finally, include your :code:`*ForCausalLM` class in `vllm/model_executor/models/__init__.py `_ and register it to the :code:`_MODEL_REGISTRY` in `vllm/model_executor/model_loader.py `_. +Finally, register your :code:`*ForCausalLM` class to the :code:`_MODELS` in `vllm/model_executor/models/__init__.py `_. 6. Out-of-Tree Model Integration --------------------------------------------