up

Author: metascroy

examples/models/llama/export_llama_lib.py

@@ -1016,7 +1016,7 @@ def _to_edge_and_lower_llama_mlx(
 
     builder = builder_exported.pt2e_quantize(quantizers).to_edge_transform_and_lower(
         partitioners,
-        passes=get_default_passes(),
+        transform_passes=get_default_passes(),
     )
 
     if verbose:

extension/llm/export/builder.py

@@ -453,7 +453,7 @@ def to_backend(self, partitioners: Optional[List[Partitioner]]) -> "LLMEdgeManag
     def to_edge_transform_and_lower(
         self,
         partitioners: Optional[List[Partitioner]],
-        passes: Optional[List] = None,
+        transform_passes: Optional[List] = None,
     ) -> "LLMEdgeManager":
         if partitioners is None:
             logging.info("No partitioner provided, skipping backend lowering...")
@@ -464,7 +464,7 @@ def to_edge_transform_and_lower(
         edge_config = self._get_edge_config()
         self.edge_manager = to_edge_transform_and_lower(
             exported_module,
-            transform_passes=passes,
+            transform_passes=transform_passes,
             partitioner=partitioners,
             compile_config=edge_config,
             constant_methods=self.metadata,