diff --git a/tests/examples/aot_mlp_test.py b/tests/examples/aot_model_test.py
similarity index 70%
rename from tests/examples/aot_mlp_test.py
rename to tests/examples/aot_model_test.py
index c4266a4af..6fe1a7319 100644
--- a/tests/examples/aot_mlp_test.py
+++ b/tests/examples/aot_model_test.py
@@ -25,7 +25,13 @@ def testMLPExportSimple(self):
     def testMLPExportSimple(self):
         _run("examples/aot_mlp/mlp_export_dynamic.py")
 
-
+class AOTLLMTest(unittest.TestCase):
+    def testParamlessModule(self):
+        command = ["python/turbine_models/custom_models/stateless_llama.py", 
+                "--compile_to=torch",
+                "--hf_auth_token=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", 
+                "--external_weights"]
+        _run(" ".join(command))
 if __name__ == "__main__":
     logging.basicConfig(level=logging.DEBUG)
     unittest.main()