diff --git a/README.md b/README.md index 0c06f4f..69d16b6 100644 --- a/README.md +++ b/README.md @@ -93,7 +93,7 @@ Save model to local with specified format. ```python from towhee import ops -op = ops.image_embedding.isc().get_op() +op = ops.image_embedding.isc(device='cpu').get_op() op.save_model('onnx', 'test.onnx') ```