a repository for PXIA models
Project description
PXIA
A repository for pxia models
This repository is using HuggingFace's PyTorchModelHubMixin classes
How to use
pip install pxia
from pxia import GPT2
model = GPT2(block_size= 1024, vocab_size = 50257, n_layer= 11, n_head= 12, n_embed = 768) # or use default parameters
model.push_to_hub("phxia/gpt2")
pretrained_model = GPT2.from_pretrained("phxia/gpt2")
alternatively you can load weights from source gpt2 models from huggingface and convert them to pxia format
model = GPT2.from_origin("openai-community/gpt2")
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pxia-0.0.5.tar.gz
(318.2 kB
view hashes)
Built Distribution
pxia-0.0.5-py3-none-any.whl
(10.2 kB
view hashes)