gpt2 / README.md

Push model using huggingface_hub.

5e5ef59 verified almost 2 years ago

783 Bytes

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