Use package_aoti API

larryliu0820 · larryliu0820 · commit b84c13b84af6 · 2024-11-01T14:55:23.000-07:00
diff --git a/torchchat/export.py b/torchchat/export.py
@@ -41,7 +41,7 @@ def export_for_server(
     package: bool = True,
 ) -> str:
     """
-    Export the model using AOT Compile to get a .dso for server use cases.
+    Export the model using AOT Compile to get a .pt2 for server use cases.
 
     Args:
         model: The model to be exported.
