add docs for LLM attr

aobo-y · aobo-y · commit e39f22ef4618 · 2023-12-04T00:35:21.000-08:00
diff --git a/sphinx/source/index.rst b/sphinx/source/index.rst
@@ -12,6 +12,7 @@ Captum API Reference
    :caption: API Reference
 
    attribution
+   llm_attr
    noise_tunnel
    layer
    neuron
diff --git a/sphinx/source/llm_attr.rst b/sphinx/source/llm_attr.rst
@@ -0,0 +1,21 @@
+LLM Attribution Classes
+========================
+
+LLMAttribution
+^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.LLMAttribution
+    :members:
+
+LLMGradientAttribution
+^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.LLMGradientAttribution
+    :members:
+
+
+LLMAttributionResult
+^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.LLMAttributionResult
+    :members:
diff --git a/sphinx/source/utilities.rst b/sphinx/source/utilities.rst
@@ -1,6 +1,18 @@
 Utilities
 ==========
 
+Interpretable Inpit
+^^^^^^^^^^^^^^
+.. autoclass:: captum.attr.InterpretableInput
+    :members:
+
+.. autoclass:: captum.attr.TextTemplateInput
+    :members:
+
+.. autoclass:: captum.attr.TextTokenInput
+    :members:
+
+
 Visualization
 ^^^^^^^^^^^^^^
 
