Fix pyre on builder.py

Differential Revision: D65837880

Author: jackzhxng

extension/llm/export/builder.py

@@ -184,8 +184,6 @@ def export(self) -> "LLMEdgeManager":
             if hasattr(self.args, "qnn") and self.args.qnn:
                 # TODO: this is temporary and export_for_training doesn't work with qnn either. We need a
                 # functional graph. See issue https://github.com/pytorch/executorch/pull/4627 for more details
-                # pyre-fixme[8]: Attribute has type `Optional[GraphModule]`; used as
-                #  `Module`.
                 exported_module = torch.export.export(
                     self.model,
                     self.example_inputs,
@@ -194,14 +192,14 @@ def export(self) -> "LLMEdgeManager":
                     strict=True,
                 )
             else:
-                # pyre-fixme[8]: Attribute has type `Optional[GraphModule]`; used as
-                #  `Module`.
                 exported_module = export_for_training(
                     self.model,
                     self.example_inputs,
                     kwargs=self.example_kwarg_inputs,
                     dynamic_shapes=dynamic_shape,
                 )
+            # pyre-fixme[8]: Attribute has type `Optional[GraphModule]`; used as
+            #  `Module`.
             self.pre_autograd_graph_module = exported_module.module()
             if hasattr(self.args, "export_only") and self.args.export_only:
                 torch.export.save(exported_module, self.args.output_name)