From 0b86a1f4e058fc24aea5818d58d3bd765742aac5 Mon Sep 17 00:00:00 2001 From: dan Date: Mon, 13 Nov 2023 00:05:37 -0600 Subject: [PATCH] add a test --- tests/examples/{aot_mlp_test.py => aot_model_test.py} | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) rename tests/examples/{aot_mlp_test.py => aot_model_test.py} (70%) 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()