Criando em modelo customizado sobre modelos do hugging face

[Kumar9 (Kumar C)]

Escrevi meu próprio modelo em PyTorch e estou tentando hospedá-lo em models, mas não estou conseguindo? Alguém tem alguma ideia?
Usei o Colab para treinar os modelos.