diff --git a/sample_model_configurations/nvidia_configs/pet.py b/sample_model_configurations/nvidia_configs/pet.py
index f59b1c4..9832cea 100644
--- a/sample_model_configurations/nvidia_configs/pet.py
+++ b/sample_model_configurations/nvidia_configs/pet.py
@@ -20,6 +20,12 @@
 CHECKPOINTS = {
     "pet-oam-xl": "pet-oam-xl@1.0.0",
     "pet-omatpes-l": "pet-omatpes-l@0.1.0",
+    "pet-mad-xs": "pet-mad-xs@1.6.0",
+    "pet-mad-s": "pet-mad-s@1.6.0",
+    "pet-mad-m": "pet-mad-m@1.6.0",
+    "pet-omol-s": "pet-omol-s@1.0.0",
+    "pet-omol-m": "pet-omol-m@1.0.0",
+    "pet-omol-l": "pet-omol-l@1.0.0",
 }