Is there a way to convert the trained model into huggingface format?

[hohoCode]

Is there a way to convert the trained model into huggingface format?

So people can just use Huggingface's AutoModel.from_pretrained("...") normally as always with your models. Easier for everything.

Thanks.

[Andron00e]

Hi @hohoCode!

Unfortunately, you cannot do this by design. To make our custom model compatible with AutoModel.from_pretrained("...") one should:

1. modify our model's class to inherit from PreTrainedModel
2. create a custom config class that inherits from PretrainedConfig
3. register your model and config

If you have found our codebase useful, feel free to submit your PR so you can help improve it)