Skip to content

Commit

Permalink
[Doc]: Update the doc of adding new models (vllm-project#4236)
Browse files Browse the repository at this point in the history
  • Loading branch information
YeFD authored and ZackBradshaw committed Apr 22, 2024
1 parent 505adb0 commit f823ced
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/models/adding_model.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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 <https://github.com/vllm-project/vllm/blob/main/vllm/model_executor/models/__init__.py>`_ and register it to the :code:`_MODEL_REGISTRY` in `vllm/model_executor/model_loader.py <https://github.com/vllm-project/vllm/blob/main/vllm/model_executor/model_loader.py>`_.
Finally, register your :code:`*ForCausalLM` class to the :code:`_MODELS` in `vllm/model_executor/models/__init__.py <https://github.com/vllm-project/vllm/blob/main/vllm/model_executor/models/__init__.py>`_.

6. Out-of-Tree Model Integration
--------------------------------------------
Expand Down

0 comments on commit f823ced

Please sign in to comment.