Fix pyre

jackzhxng · jackzhxng · commit a73f1073e652 · 2025-02-25T14:18:29.000-08:00
diff --git a/examples/models/llama/export_llama_lib.py b/examples/models/llama/export_llama_lib.py
@@ -608,7 +608,8 @@ def _prepare_for_llama_export(args) -> LLMEdgeManager:
     )
 
     _set_quantized_computation_dtype(
-        edge_manager.model, dtype_override.to_torch_dtype()
+        edge_manager.model,
+        dtype_override.to_torch_dtype(),  # pyre-ignore[16]
     )
 
     return edge_manager

Original file line number	Diff line number	Diff line change
`@@ -608,7 +608,8 @@ def _prepare_for_llama_export(args) -> LLMEdgeManager:`
`608`	`608`	`)`
`609`	`609`
`610`	`610`	`_set_quantized_computation_dtype(`
`611`		`- edge_manager.model, dtype_override.to_torch_dtype()`
	`611`	`+ edge_manager.model,`
	`612`	`+ dtype_override.to_torch_dtype(), # pyre-ignore[16]`
`612`	`613`	`)`
`613`	`614`
`614`	`615`	`return edge_manager`