search

huggingface / swift-coreml-transformers

Swift Core ML 3 implementations of GPT-2, DistilGPT-2, BERT, and DistilBERT for Question answering. Other Transformers coming soon!
Apache License 2.0
1.62k stars 179 forks source link

GPT2-large #35

Closed kpacholak closed 1 year ago

kpacholak commented 1 year ago

Hi! Do you know, how can I get gpt2-large version? is it as simple as change model_name = "gpt2" to model_name = "gpt2-large" in Python converting script?

julien-c commented 1 year ago

you should use this newer tool: https://huggingface.co/spaces/huggingface-projects/transformers-to-coreml

cc @pcuenca