Lint

jackzhxng · jackzhxng · commit e07f810d085a · 2025-02-24T16:54:16.000-08:00
diff --git a/examples/models/llama/export_llama_lib.py b/examples/models/llama/export_llama_lib.py
@@ -605,7 +605,9 @@ def _prepare_for_llama_export(args) -> LLMEdgeManager:
         )
     )
 
-    _set_quantized_computation_dtype(edge_manager.model, dtype_override.to_torch_dtype())
+    _set_quantized_computation_dtype(
+        edge_manager.model, dtype_override.to_torch_dtype()
+    )
 
     return edge_manager
 

Original file line number	Diff line number	Diff line change
`@@ -605,7 +605,9 @@ def _prepare_for_llama_export(args) -> LLMEdgeManager:`
`605`	`605`	`)`
`606`	`606`	`)`
`607`	`607`
`608`		`- _set_quantized_computation_dtype(edge_manager.model, dtype_override.to_torch_dtype())`
	`608`	`+ _set_quantized_computation_dtype(`
	`609`	`+ edge_manager.model, dtype_override.to_torch_dtype()`
	`610`	`+ )`
`609`	`611`
`610`	`612`	`return edge_manager`
`611`	`613`