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")
we also support AutoModel class in our repository
to load any pxia model either locally or from the hub you can
from pxia import AutoModel
model = AutoModel.from_pretrained("phxia/gpt2")
1 maintainer
Metadata
1 maintainer
Release files for pxia 0.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| pxia-0.2.0.tar.gz | 319.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pxia-0.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 332.4 kB
Release files / pxia-0.2.0.tar.gz
| Download URL | pxia-0.2.0.tar.gz |
|---|---|
| Size | 319.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
b85684c1460906de6ab346ca590ba8102257e83fbeb76c4b144d7c9516920a04
|
|
BLAKE2b-256 checksum How to use checksums |
3dc4be80b8d237c84973e8ccc521d7c2a585edc5b69ba7778261f2b8e9abacc1
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.9.20
|
Release files / pxia-0.2.0-py3-none-any.whl
| Download URL | pxia-0.2.0-py3-none-any.whl |
|---|---|
| Size | 12.4 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
1288196c207f11bdb2d69e29d7736edea298604a324f3bc5d02a9f0ca93d3e78
|
|
BLAKE2b-256 checksum How to use checksums |
88a549fa95737f1f4d64bc515a07c730273fbe582c23987e132323ee468eb889
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.9.20
|
