Hello @jplu
I tested TFdistilbertModel but after conversion, it’s only expecting tensor of 3*5 dimension. The output of the tokenizer is a single-dimensional array.
I also try to set input like this
input_spec = tf.TensorSpec([1, 768], tf.int32)
model._set_inputs(input_spec, training=False)

But when I check model.inputs it was None and  I was getting same error related to dimension.