remove nocommits

pytorch · May 6, 2024 · 2d7c574 · 2d7c574
1 parent 5071bec
commit 2d7c574
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/export_aoti.py b/export_aoti.py
@@ -26,9 +26,6 @@ def export_model(model: nn.Module, device, output_path, args=None):
 
     model.to(device)
 
-    from export_aoti_util import replace_linear_with_custom_linear # NOCOMMIT
-    replace_linear_with_custom_linear(model) # NOCOMMIT
-
     so = torch._export.aot_compile(
         model,
         args=input,