sunlabuiuc · fbonc · Aug 26, 2026 · Aug 28, 2026
diff --git a/docs/api/models.rst b/docs/api/models.rst
@@ -201,7 +201,6 @@ API Reference
     models/pyhealth.models.TCN
     models/pyhealth.models.TFMTokenizer
     models/pyhealth.models.GAN
-    models/pyhealth.models.VAE
     models/pyhealth.models.HALO
     models/pyhealth.models.GPT2
     models/pyhealth.models.PromptEHR

diff --git a/docs/api/models/pyhealth.models.VAE.rst b/docs/api/models/pyhealth.models.VAE.rst
diff --git a/docs/tutorials.rst b/docs/tutorials.rst
@@ -214,8 +214,6 @@ These examples are located in ``examples/cxr/``.
      - Multi-label classification on ChestX-ray14 dataset (notebook)
    * - ``cxr/ChestXrayClassificationWithSaliency.ipynb``
      - Chest X-ray classification with saliency maps (notebook)
-   * - ``cxr/chextXray_image_generation_VAE.py``
-     - VAE for chest X-ray image generation
    * - ``cxr/ChestXray-image-generation-GAN.ipynb``
      - GAN for chest X-ray image generation (notebook)
 
@@ -328,8 +326,6 @@ Notebooks (Interactive)
      - Language model embeddings with OpenAI
    * - ``prepare_mapping.ipynb``
      - Data preprocessing and mapping utilities
-   * - ``graph_torchvision_model.ipynb``
-     - Using Torchvision models with graph data
 
 
 ----------
diff --git a/examples/cxr/chextXray_image_generation_VAE.py b/examples/cxr/chextXray_image_generation_VAE.py