diff --git a/llama-cpp-2/Cargo.toml b/llama-cpp-2/Cargo.toml
index b10ebd60..97cd612c 100644
--- a/llama-cpp-2/Cargo.toml
+++ b/llama-cpp-2/Cargo.toml
@@ -32,3 +32,6 @@ sampler = []
 
 [lints]
 workspace = true
+
+[package.metadata.docs.rs]
+features = ["sampler"]
\ No newline at end of file