remove extraneous call to wrapper

pytorch · Apr 7, 2024 · c78662f · c78662f
1 parent 13697fa
commit c78662f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/export_et.py b/export_et.py
@@ -73,11 +73,12 @@ def materialze_broadcast_of_rope_freq_cis(
 def canonical_path(path):
     return path
 
-## align AOTI and ET export
-# def export_model(model: nn.Module, device, output_path):
+
 def export_model(model, device, output_path, args=None) -> str:  # noqa: C901
 
-    export_model = model_wrapper(model, device=device)
+    # applied wrapper already in export.
+    # export_model = model_wrapper(model, device=device)
+    export_model = model
     print(export_model)
 
     input = (