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
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.2.0.tar.gz
(319.9 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
pxia-0.2.0-py3-none-any.whl
(12.4 kB
view details)
File details
Details for the file pxia-0.2.0.tar.gz.
File metadata
- Download URL: pxia-0.2.0.tar.gz
- Upload date:
- Size: 319.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/5.1.1 CPython/3.9.20
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b85684c1460906de6ab346ca590ba8102257e83fbeb76c4b144d7c9516920a04
|
|
| MD5 |
8e557db68d0cdc87a27ddb49b022e96f
|
|
| BLAKE2b-256 |
3dc4be80b8d237c84973e8ccc521d7c2a585edc5b69ba7778261f2b8e9abacc1
|
File details
Details for the file pxia-0.2.0-py3-none-any.whl.
File metadata
- Download URL: pxia-0.2.0-py3-none-any.whl
- Upload date:
- Size: 12.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/5.1.1 CPython/3.9.20
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1288196c207f11bdb2d69e29d7736edea298604a324f3bc5d02a9f0ca93d3e78
|
|
| MD5 |
993c8b5b7557de7ce2cb79d17f62c4e7
|
|
| BLAKE2b-256 |
88a549fa95737f1f4d64bc515a07c730273fbe582c23987e132323ee468eb889
|
