Make it runnable via buck

Differential Revision: D61567500

Pull Request resolved: #4811

Author: helunwencser

examples/models/phi-3-mini/export_phi-3-mini.py

@@ -28,7 +28,7 @@
 from .phi_3_mini import Phi3Mini
 
 
-def main(args) -> None:
+def export(args) -> None:
     torch.manual_seed(0)
 
     if args.context_length == "4k":
@@ -90,7 +90,7 @@ def main(args) -> None:
         file.write(et_program.buffer)
 
 
-if __name__ == "__main__":
+def main():
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "-c",
@@ -113,4 +113,8 @@ def main(args) -> None:
         default="phi-3-mini.pte",
         help="Override the output filename of the saved pte model file.",
     )
-    main(parser.parse_args())
+    export(parser.parse_args())
+
+
+if __name__ == "__main__":
+    main()