cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change supported_ops to a Tuple instead of List to match the API definition.

Browse Source 
PiperOrigin-RevId: 557890361
This commit is contained in:
MediaPipe Team 2023-08-17 11:46:32 -07:00 committed by Copybara-Service
parent 990bfd2e3e
commit b213256cbd
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mediapipe/model_maker/python/vision/face_stylizer/face_stylizer.py
View File

@ -241,10 +241,11 @@ class FaceStylizer(object):
face_stylizer_model_buffer = model_util.convert_to_tflite( face_stylizer_model_buffer = model_util.convert_to_tflite(
model=model, model=model,
supported_ops=[ quantization_config=None,
supported_ops=(
tf.lite.OpsSet.TFLITE_BUILTINS, tf.lite.OpsSet.TFLITE_BUILTINS,
tf.lite.OpsSet.SELECT_TF_OPS, tf.lite.OpsSet.SELECT_TF_OPS,
], ),
preprocess=self._preprocessor, preprocess=self._preprocessor,
) )