Wheelodex » Projects » fms-hf-tuning

fms-hf-tuning

View on PyPI — Reverse Dependencies (0)

0.0.1 fms_hf_tuning-0.0.1-py3-none-any.whl

Wheel Details

Project: fms-hf-tuning
Version: 0.0.1
Filename: fms_hf_tuning-0.0.1-py3-none-any.whl
Download: [link]
Size: 34701
MD5: e220a4e373200bdae124c65f873a0204
SHA256: f716a386db33937cd57feffe48fd7575c4fbfb93fdd5cd63a231ac998f530b81
Uploaded: 2024-04-03 18:54:51 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: fms-hf-tuning
Version: 0.0.1
Summary: FMS HF Tuning
Author-Email: Sukriti Sharma <sukriti.sharma4[at]ibm.com>, Anh Uong <anh.uong[at]ibm.com>
Project-Url: Homepage, https://github.com/foundation-model-stack/fms-hf-tuning
Project-Url: Repository, https://github.com/foundation-model-stack/fms-hf-tuning
Project-Url: Issues, https://github.com/foundation-model-stack/fms-hf-tuning/issues
License: Apache-2.0
Keywords: fms-hf-tuning,python,tuning
Classifier: License :: OSI Approved :: Apache Software License
Requires-Python: ~=3.9
Requires-Dist: numpy
Requires-Dist: accelerate (>=0.20.3)
Requires-Dist: transformers
Requires-Dist: torch
Requires-Dist: sentencepiece
Requires-Dist: tokenizers (>=0.13.3)
Requires-Dist: tqdm
Requires-Dist: trl
Requires-Dist: peft (>=0.8.0)
Requires-Dist: datasets (>=2.15.0)
Requires-Dist: fire
Requires-Dist: aim (==3.18.1); extra == "aim"
Requires-Dist: wheel; extra == "dev"
Requires-Dist: packaging; extra == "dev"
Requires-Dist: ninja; extra == "dev"
Requires-Dist: flash-attn; extra == "flash-attn"
Provides-Extra: aim
Provides-Extra: dev
Provides-Extra: flash-attn
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 13062 characters]

WHEEL

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

RECORD

Path Digest Size
tests/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tests/helpers.py sha256=fWvJPofomqoIohj8RXD39_M7SJU7RXvv4ZOj1g4TFX4 1432
tests/test_sft_trainer.py sha256=SCCmG_w0qoS3sGDa-s8kafcupaVSaV10lpLB0898pV8 11750
tests/build/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tests/build/dummy_job_config.json sha256=Oo6Wi_pbL8ICR8xHKtARj-1vg8LU4Kgbbg6xVeQRfPE 1164
tests/build/test_utils.py sha256=-9ZQzUaBnLRGAXZ_7JB5LaY_QfJp0VFPqJWUWMJZynk 2780
tests/data/__init__.py sha256=L6YdnL0BsUazoYiqGa5lrU7z0zjK1GjGXuzsWJXFjVk 831
tests/data/twitter_complaints_small.json sha256=7s6_QbTwd1s93UIfRCBQNgwMd1HLGs2CZDfMcaKuBcw 3198
tests/utils/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tests/utils/test_data_type_utils.py sha256=R5dnQY_OQBEMShM0g_OAigw8JRiRhdRZZDONRRfyvaY 1535
tuning/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/aim_loader.py sha256=9wGbemgKySydJwwZavdlX8ywp4WFN4H8r-Pbeg0V1iw 1384
tuning/sft_trainer.py sha256=tHwNx9neSVcZNnmKe_JfBgskDp93ojR6HUEm1wsLU3w 11268
tuning/config/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/config/configs.py sha256=5J0Vg_H35WWPgOddFUugms4P7S1dpGI_5CYHf2W8ZTk 3077
tuning/config/peft_config.py sha256=28Wp7eNpeJvM-DaWH6yUOarTL0-5GpiSN-GWDQ1hPIk 1463
tuning/data/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/data/tokenizer_data_utils.py sha256=PlcBJQm7jgLmUqAPzBuka0O3GaRPmhhcOTZPnewBrXg 1593
tuning/utils/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/utils/config_utils.py sha256=kp3SZETwPFd_xJY537MSCk1vUJWqCyjuaz-Q7HTkWBA 3135
tuning/utils/data_type_utils.py sha256=LHP6V8CQUkvYImTTS-bOHj_Ni4_hsJiyr_s_qJlS9jg 2006
tuning/utils/import_utils.py sha256=0XoH5joeg4ewEPjDCr4J7vW9LoqtLF-Pth42VwVcEBk 770
tuning/utils/merge_model_utils.py sha256=PoSwZXmnlQICOFRSbPUkVGu3vJFTnh7BQGHoPWr9VsM 4136
fms_hf_tuning-0.0.1.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
fms_hf_tuning-0.0.1.dist-info/METADATA sha256=daGQ_-HL-5T6l_McpNRJrnFUgLZJxKuEMj0d4MFU_Fg 14248
fms_hf_tuning-0.0.1.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
fms_hf_tuning-0.0.1.dist-info/top_level.txt sha256=LHHl3cBlt-qGTq8JS6GRex6bXixEq5WCe8sv3QEIL8s 13
fms_hf_tuning-0.0.1.dist-info/RECORD — —

top_level.txt

tests
tuning