From 2c0fec9a56baadb56ed9a7166e9267bdec68a4ce Mon Sep 17 00:00:00 2001
From: Jack Zhang <32371937+jackzhxng@users.noreply.github.com>
Date: Mon, 3 Mar 2025 11:29:19 -0800
Subject: [PATCH] Add previous logging for to_edge model in export_llama

---
 extension/llm/export/builder.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/extension/llm/export/builder.py b/extension/llm/export/builder.py
index 47ad30e9390..eb8dd462378 100644
--- 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(