huggingface / exporters

Export Hugging Face models to Core ML and TensorFlow Lite
Apache License 2.0
622 stars 46 forks source link

Support for `Voicelab/vlt5-base-keywords` #73

Closed underthestars-zhy closed 9 months ago

underthestars-zhy commented 9 months ago

I got a problem when I want to convert a vlt5 model to coreml model

KeyError: "voicelab/vlt5-base-keywords is not supported yet. Only ['bart', 'beit', 'bert', 'big_bird', 'bigbird_pegasus', 'blenderbot', 'blenderbot_small', 'bloom', 'convnext', 'ctrl', 'cvt', 'data2vec', 'distilbert', 'ernie', 'falcon', 'gpt2', 'gpt_bigcode', 'gptj', 'gpt_neo', 'gpt_neox', 'levit', 'llama', 'm2m_100', 'marian', 'mistral', 'mobilebert', 'mobilevit', 'mobilevitv2', 'mvp', 'pegasus', 'plbart', 'roberta', 'roformer', 'segformer', 'splinter', 'squeezebert', 't5', 'vit', 'yolos'] are supported. If you want to support voicelab/vlt5-base-keywords please propose a PR or open up an issue."