Skip to content

Commit d43b22a

Browse files
authored
Update build_rknn.py
1 parent e5452f8 commit d43b22a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

build_rknn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def ConvertModel(model_path='ViT-B-32__openai/textual/model.onnx', target_platfo
7272
modified = modify_onnx_change_op_type(model_path, modified_onnx_path, "CumSum", "CPUCumSum")
7373
onnx_to_load = modified_onnx_path if modified else model_path
7474
if modified:
75-
ret = rknn.reg_custom_op(RKCumSum())
75+
ret = rknn.reg_custom_op(CPUCumSum())
7676

7777
if ret != 0:
7878
raise RuntimeError("Register Custom OP failed!")

0 commit comments

Comments
 (0)