| 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](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration. | |
| Library: [pxia](https://github.com/not-lain/pxia) | |
| ## how to load | |
| ``` | |
| pip install pxia | |
| ``` | |
| use the AutoModel class | |
| ```python | |
| from pxia AutoModel | |
| model = AutoModel.from_pretrained("phxia/gpt2") | |
| ``` | |
| or you can use the model class directly | |
| ```python | |
| from pxia import GPT2 | |
| model = GPT2.from_pretrained("phxia/gpt2") | |
| ``` | |
| ## Contributions | |
| Any contributions are welcome at https://github.com/not-lain/pxia | |
| <img src="https://huggingface.co/spaces/phxia/README/resolve/main/logo.png"/> | |