llama-trainer

View on PyPIReverse Dependencies (0)

0.2.1 llama_trainer-0.2.1-py3-none-any.whl

Wheel Details

Project: llama-trainer
Version: 0.2.1
Filename: llama_trainer-0.2.1-py3-none-any.whl
Download: [link]
Size: 11535
MD5: faa292397d747a76b474981841cc9958
SHA256: 03caffce8f476cbdea06f280e7a84fc55bc037b3489cce1ff3184a285b64afea
Uploaded: 2023-09-28 08:48:02 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: llama-trainer
Version: 0.2.1
Summary: Llama trainer utility
Author: Riccardo Orlando
Author-Email: orlandorcc[at]gmail.com
Home-Page: https://github.com/Riccorl/llama-trainer
License: Apache
Keywords: NLP deep learning transformer pytorch llama llms hf huggingface
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: torch (<2.1,>=1.10)
Requires-Dist: accelerate (<0.23,>=0.21)
Requires-Dist: peft (<0.5,>=0.4)
Requires-Dist: bitsandbytes (==0.40.2)
Requires-Dist: transformers (<5.1,>=4.31)
Requires-Dist: trl (==0.4.7)
Requires-Dist: scipy
Requires-Dist: black
Description-Content-Type: text/markdown
[Description omitted; length: 1120 characters]

WHEEL

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

RECORD

Path Digest Size
llama_trainer/__init__.py sha256=LNhCpEUO03bzkqaA3D9WTBeQ89rbWupfz4xY1-C0bYg 90
llama_trainer/infer.py sha256=VpK9lSoQw86RLFtbXZ-MLRXXpbuI2iYMMG88U917afY 2988
llama_trainer/trainer.py sha256=QNf_CbGv23CS0H-KtPlMKmlC01gvHbx1n1WQOZapqiE 9356
llama_trainer/version.py sha256=kkrcUyJ4qz054WOcSnhNLvqqTwZb0ipOYLib3JzMA6A 466
llama_trainer/cli/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
llama_trainer/cli/train.py sha256=H6AvaSA2s-x0AOOMrmiJ7rlhSoc3cHuUHc4uDo84uWs 5218
llama_trainer/utils/__init__.py sha256=xLk1RhwOYDFCMMewa4CYtYPlJ7_0NxfaCdQV3qidwxU 262
llama_trainer/utils/llama_patch.py sha256=dubZer3mYfZv0ZoWzs9-MouSaA6Fne_aqGnFX5SGH5Y 6147
llama_trainer/utils/stopping_criteria.py sha256=28-rSJj1v2qSrdv4ULhEFIKjqwB9jwj9pP9MfSiJCHw 906
llama_trainer-0.2.1.dist-info/METADATA sha256=miVCVmRoJLgORbi3X9SureJfHtbupuo6ulK4Ym4bRug 1959
llama_trainer-0.2.1.dist-info/WHEEL sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A 92
llama_trainer-0.2.1.dist-info/top_level.txt sha256=8k1VVgmrHsSRfUkfDTDuWUh5sJy3sDmhRMbjLLqpl28 14
llama_trainer-0.2.1.dist-info/RECORD

top_level.txt

llama_trainer