Skip to content
This repository was archived by the owner on Jan 10, 2025. It is now read-only.

Commit 37976c4

Browse files
PierrciPierrci
committed
update distilbert script
1 parent 92c8e4b commit 37976c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎models_generation/distilbert.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
print(model.outputs)
1111

1212
converter = tf.lite.TFLiteConverter.from_keras_model(model)
13-
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS, tf.lite.OpsSet.SELECT_TF_OPS]
13+
converter.target_spec.supported_ops = [tf.lite.OpsSet.SELECT_TF_OPS]
1414

1515
tflite_model = converter.convert()
1616

0 commit comments

Comments
 (0)