File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -38,12 +38,11 @@ def preprocess(
3838 PreprocessResult: The result of preprocessing, including the compiled model bytes.
3939 """
4040 # Apply RemoveCloneOpsTransform to eliminate unnecessary clone operations
41- remove_clone_transform = RemoveCloneOpsTransform ()
42- transformed_result = remove_clone_transform (edge_program .graph_module )
41+ transformed_ep = RemoveCloneOpsTransform ()(edge_program .graph_module )
4342
4443 # Update the edge_program with the transformed graph
45- if transformed_result .graph_module is not None :
46- edge_program ._graph_module = transformed_result .graph_module
44+ if transformed_ep .graph_module is not None :
45+ edge_program ._graph_module = transformed_ep .graph_module
4746
4847 input_names = edge_program .graph_signature .user_inputs
4948 args = []
You can’t perform that action at this time.
0 commit comments