Add previous logging for to_edge model in export_llama (#8889)

jackzhxng · web-flow · commit e8d20090364c · 2025-03-03T13:14:26.000-08:00
diff --git a/extension/llm/export/builder.py b/extension/llm/export/builder.py
@@ -448,6 +448,8 @@ def to_edge_transform_and_lower(
             compile_config=edge_config,
             constant_methods=self.metadata,
         )
+        if self.verbose:
+            logging.info(f"Exported graph:\n{self.edge_manager.exported_program()}")
         return self
 
     def to_executorch(

Original file line number	Diff line number	Diff line change
`@@ -448,6 +448,8 @@ def to_edge_transform_and_lower(`
`448`	`448`	`compile_config=edge_config,`
`449`	`449`	`constant_methods=self.metadata,`
`450`	`450`	`)`
	`451`	`+ if self.verbose:`
	`452`	`+ logging.info(f"Exported graph:\n{self.edge_manager.exported_program()}")`
`451`	`453`	`return self`
`452`	`454`
`453`	`455`	`def to_executorch(`