huggingface / tflite-android-transformers

DistilBERT / GPT-2 for on-device inference thanks to TensorFlow Lite with Android demo apps
Apache License 2.0
395 stars 81 forks source link

Any ideas on how we can convert a model from huggingface (transformers library )to tensorflow lite? #17

Open BENSAFOUAN-Abdelhalim opened 2 years ago

BENSAFOUAN-Abdelhalim commented 2 years ago

I want to convert this model to tensoflow lite i download it from here, because when i want to save the model locally it gives me the model with 'bin' format.

when i try to convert the model it gives me this error : AttributeError: 'CamembertForQuestionAnswering' object has no attribute 'call' by using tf_model.h5 file. Also i can't load it using : tf.keras.models.load_model() it gives me : ValueError: No model config found in the file at <tensorflow.python.platform.gfile.GFile object at 0x7f27cceb1810>. when i want to save the transformers model locally it gives me the model with 'bin' format, so i download it from the platform.