diff --git a/docs/index.md b/docs/index.md
index 5424e26..c36adff 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -105,12 +105,16 @@ python3 setup.py develop
             - __call__
             - create_chat_completion
             - set_cache
+            - save_state
+            - load_state
             - token_bos
             - token_eos
         show_root_heading: true
 
 ::: llama_cpp.LlamaCache
 
+::: llama_cpp.LlamaState
+
 ::: llama_cpp.llama_cpp
     options:
         show_if_no_docstring: true