remove exir.capture from models.py (#2325)

Summary:
Pull Request resolved: #2325

capture is deprecated, use export

bypass-github-pytorch-ci-checks

Reviewed By: mergennachin

Differential Revision: D54695851

fbshipit-source-id: fd23fce13a29cb76c8964ab009e8c72eef48e5e0

Author: JacobSzwejbka

exir/tests/models.py

@@ -13,8 +13,10 @@
 
 import torch  # noqa: F401
 import torch.nn as nn
+from executorch.exir import to_edge
 from executorch.exir.lowered_backend_module import LoweredBackendModule
 from torch import Tensor
+from torch.export import export
 
 # TODO: add one more test for data dependent op plus repeat
 
@@ -148,13 +150,14 @@ def get_random_inputs(self) -> Tuple[Tensor, Tensor]:
                 return (torch.randn(1, 3), torch.randn(1, 3))
 
         delegated_m = DelegateAdd()
-        edge_ir_m = exir.capture(
-            delegated_m,
-            delegated_m.get_random_inputs(),
-            exir.CaptureConfig(),
-        ).to_edge()
+        edge_ir_m = to_edge(
+            export(
+                delegated_m,
+                delegated_m.get_random_inputs(),
+            )
+        )
         lowered_module = LoweredBackendModule(
-            edge_program=edge_ir_m.exported_program,
+            edge_program=edge_ir_m.exported_program(),
             backend_id="backend_demo",
             processed_bytes=bytes("basic_module_add", encoding="utf8"),
             compile_specs=[],