Merge pull request google-ai-edge#687 from gudgud96:fix/export-directory

PiperOrigin-RevId: 764814512

Author: copybara-github

ai_edge_torch/model.py

@@ -155,7 +155,8 @@ def export(self, path: str) -> None:
     Args:
       path: The path to file to which the model is serialized.
     """
-    os.makedirs(os.path.dirname(path), exist_ok=True)
+    if os.path.dirname(path):
+      os.makedirs(os.path.dirname(path), exist_ok=True)
     with open(path, 'wb') as file_handle:
       file_handle.write(self._tflite_model)