metadata
library_name: pxia
tags:
- gpt2
- model_hub_mixin
- pxia
- pytorch_model_hub_mixin
- text-generation
This model has been pushed to the Hub using the PytorchModelHubMixin integration.
Library: pxia
how to load
pip install pxia
use the AutoModel class
from pxia AutoModel
model = AutoModel.from_pretrained("phxia/gpt2")
or you can use the model class directly
from pxia import GPT2
model = GPT2.from_pretrained("phxia/gpt2")
Contributions
Any contributions are welcome at https://github.com/not-lain/pxia
