Update pix2tex.rst

yyalon · May 20, 2022 · d1f21f9 · d1f21f9
1 parent a0d6b3d
commit d1f21f9
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/docs/pix2tex.rst b/docs/pix2tex.rst
@@ -106,7 +106,18 @@ pix2tex.dataset.scraping module
 pix2tex.models package
 ----------------------
 
-.. automodule:: pix2tex.models
+pix2tex.models.hybrid module
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. automodule:: pix2tex.models.hybrid
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
+
+pix2tex.models.vit module
+~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. automodule:: pix2tex.models.vit
    :members:
    :no-undoc-members:
    :show-inheritance: