From 9e7193d36f13d182bfba81b9649dea16dbfe9128 Mon Sep 17 00:00:00 2001 From: Glimmer of Code Date: Mon, 14 Sep 2026 18:48:56 -0700 Subject: [PATCH] Include more scientifically relevant PET models --- sample_model_configurations/nvidia_configs/pet.py | 6 ++++++ 1 file changed, 6 insertions(+) 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", }