axolotl

View on PyPIReverse Dependencies (0)

0.5.0 axolotl-0.5.0-py3-none-any.whl

Wheel Details

Project: axolotl
Version: 0.5.0
Filename: axolotl-0.5.0-py3-none-any.whl
Download: [link]
Size: 6095
MD5: 0e9d7d18a0ad8e29dc26304f3938f767
SHA256: ba9316082f0a271c4d4746bc8205678e3fdddf53e02a0dd1fb9ee413ae8584cb
Uploaded: 2024-11-08 19:51:38 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: axolotl
Version: 0.5.0
Summary: LLM Trainer
Requires-Dist: packaging (==23.2)
Requires-Dist: peft (==0.13.2)
Requires-Dist: transformers (==4.46.1)
Requires-Dist: tokenizers (>=0.20.1)
Requires-Dist: bitsandbytes (==0.44.1)
Requires-Dist: accelerate (==1.1.0)
Requires-Dist: datasets (==3.0.1)
Requires-Dist: pydantic (==2.6.3)
Requires-Dist: addict
Requires-Dist: fire
Requires-Dist: PyYAML (>=6.0)
Requires-Dist: requests
Requires-Dist: sentencepiece
Requires-Dist: wandb
Requires-Dist: einops
Requires-Dist: optimum (==1.16.2)
Requires-Dist: hf-transfer
Requires-Dist: colorama
Requires-Dist: numba
Requires-Dist: numpy (<=2.0.1,>=1.24.4)
Requires-Dist: evaluate (==0.4.1)
Requires-Dist: scipy
Requires-Dist: scikit-learn (==1.4.2)
Requires-Dist: pynvml
Requires-Dist: art
Requires-Dist: gradio (==3.50.2)
Requires-Dist: tensorboard
Requires-Dist: python-dotenv (==1.0.1)
Requires-Dist: autoawq (>=0.2.5)
Requires-Dist: triton (>=2.3.0)
Requires-Dist: liger-kernel (==0.4.0)
Requires-Dist: s3fs (>=2024.5.0)
Requires-Dist: gcsfs (>=2024.5.0)
Requires-Dist: trl (==0.12.0)
Requires-Dist: zstandard (==0.22.0)
Requires-Dist: fastcore
Requires-Dist: lm-eval (==0.4.4)
Requires-Dist: langdetect (==1.0.9)
Requires-Dist: immutabledict (==4.2.0)
Requires-Dist: antlr4-python3-runtime (==4.13.2)
Requires-Dist: torch (==2.3.1)
Requires-Dist: xformers (>=0.0.27)
Requires-Dist: auto-gptq (==0.5.1); extra == "auto-gptq"
Requires-Dist: deepspeed (==0.14.4); extra == "deepspeed"
Requires-Dist: deepspeed-kernels; extra == "deepspeed"
Requires-Dist: flash-attn (==2.6.3); extra == "flash-attn"
Requires-Dist: galore-torch; extra == "galore"
Requires-Dist: lion-pytorch (==0.1.2); extra == "lion-pytorch"
Requires-Dist: mamba-ssm (==1.2.0.post1); extra == "mamba-ssm"
Requires-Dist: causal-conv1d; extra == "mamba-ssm"
Requires-Dist: mlflow; extra == "mlflow"
Requires-Dist: galore-torch; extra == "optimizers"
Requires-Dist: lion-pytorch (==0.1.2); extra == "optimizers"
Requires-Dist: lomo-optim (==0.1.1); extra == "optimizers"
Requires-Dist: torch-optimi (==0.2.1); extra == "optimizers"
Provides-Extra: auto-gptq
Provides-Extra: deepspeed
Provides-Extra: flash-attn
Provides-Extra: galore
Provides-Extra: lion-pytorch
Provides-Extra: mamba-ssm
Provides-Extra: mlflow
Provides-Extra: optimizers
License-File: LICENSE
[Description omitted; length: 143 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.44.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
axolotl-0.5.0.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
axolotl-0.5.0.dist-info/METADATA sha256=_1tF2AA91Pi5Dw3V_LlF03suYNvt4ZR_vUa5thkktas 2377
axolotl-0.5.0.dist-info/WHEEL sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc 92
axolotl-0.5.0.dist-info/dependency_links.txt sha256=pOh7t6LPpqvpAHuMdJR9MlmC8K-q2tm9DbIlNn-UqnE 56
axolotl-0.5.0.dist-info/top_level.txt sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
axolotl-0.5.0.dist-info/RECORD

top_level.txt

[empty]
https://huggingface.github.io/autogptq-index/whl/cu118/