add a test

nod-ai · Nov 13, 2023 · 0b86a1f · 0b86a1f
1 parent e631023
commit 0b86a1f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/examples/aot_mlp_test.py → tests/examples/aot_model_test.py b/tests/examples/aot_mlp_test.py → 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()