up

metascroy · metascroy · commit 9e9ae4e7acb2 · 2025-06-10T17:27:16.000-07:00
diff --git a/examples/models/llava/export_llava.py b/examples/models/llava/export_llava.py
@@ -256,6 +256,7 @@ def export_all(llava_model: LlavaModel):
             do_quant_fusion_and_const_prop=True,
         )
     )
+    logging.info("TOKEN EMBEDDING PROG", executorch_program.exported_program("token_embedding"))
     for execution_plan in executorch_program._emitter_output.program.execution_plan:
         logging.info(
             f"Required memory for activation in bytes: {execution_plan.non_const_buffer_sizes}"

Original file line number	Diff line number	Diff line change
`@@ -256,6 +256,7 @@ def export_all(llava_model: LlavaModel):`
`256`	`256`	`do_quant_fusion_and_const_prop=True,`
`257`	`257`	`)`
`258`	`258`	`)`
	`259`	`+ logging.info("TOKEN EMBEDDING PROG", executorch_program.exported_program("token_embedding"))`
`259`	`260`	`for execution_plan in executorch_program._emitter_output.program.execution_plan:`
`260`	`261`	`logging.info(`
`261`	`262`	`f"Required memory for activation in bytes: {execution_plan.non_const_buffer_sizes}"`