Fix test

jackzhxng · jackzhxng · commit a28ddabd1873 · 2025-03-28T11:06:37.000-07:00
diff --git a/examples/models/llama/export_llama_lib.py b/examples/models/llama/export_llama_lib.py
@@ -905,7 +905,7 @@ def _to_edge_and_lower_llama_xnnpack(
     return builder.to_executorch(passes=additional_passes)
 
 
-def _to_edge_and_lower_llama(
+def _to_edge_and_lower_llama(  # noqa: C901
     builder_exported,
     modelname,
     additional_passes,
diff --git a/examples/models/llava/export_llava.py b/examples/models/llava/export_llava.py
@@ -162,7 +162,7 @@ def forward(self, images):
             use_kv_cache=True,
             example_inputs=(resized,),
             dynamic_shapes=dynamic_shapes,
-            args=None,
+            config=None,
         )
         .export()
         .pt2e_quantize([quantizer])

Original file line number	Diff line number	Diff line change
`@@ -162,7 +162,7 @@ def forward(self, images):`
`162`	`162`	`use_kv_cache=True,`
`163`	`163`	`example_inputs=(resized,),`
`164`	`164`	`dynamic_shapes=dynamic_shapes,`
`165`		`- args=None,`
	`165`	`+ config=None,`
`166`	`166`	`)`
`167`	`167`	`.export()`
`168`	`168`	`.pt2e_quantize([quantizer])`