Skip to content

Commit d2c011e

Browse files
authored
Fix extract model script (#15924)
deserialize_pte_binary now returns an object with .program as a field, instead of returning program directly.
1 parent 92bf722 commit d2c011e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/apple/coreml/scripts/extract_coreml_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222

2323
def extract_coreml_models(pte_data: bytes):
24-
program = deserialize_pte_binary(pte_data)
24+
program = deserialize_pte_binary(pte_data).program
2525
delegates: List[BackendDelegate] = sum(
2626
[execution_plan.delegates for execution_plan in program.execution_plan], []
2727
)

0 commit comments

Comments
 (0)