Wheelodex » Projects » fms-hf-tuning

fms-hf-tuning

View on PyPI — Reverse Dependencies (0)

2.1.1 fms_hf_tuning-2.1.1-py3-none-any.whl

Wheel Details

Project: fms-hf-tuning
Version: 2.1.1
Filename: fms_hf_tuning-2.1.1-py3-none-any.whl
Download: [link]
Size: 84498
MD5: 940d9cf50c1c7cbfcb6d92e0f50a53c6
SHA256: 17c8f5148ebd9e55cada0afac3b44e30c316a40a0208522d6968d87ff4b843cd
Uploaded: 2024-11-06 00:19:22 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: fms-hf-tuning
Version: 2.1.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
Classifier: Development Status :: 4 - Beta
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: ~=3.9
Requires-Dist: numpy (<2.0,>=1.26.4)
Requires-Dist: accelerate (!=0.34,<1.1,>=0.20.3)
Requires-Dist: transformers (<4.46,>4.41)
Requires-Dist: torch (<2.5,>=2.2.0)
Requires-Dist: sentencepiece (<0.3,>=0.1.99)
Requires-Dist: tokenizers (<1.0,>=0.13.3)
Requires-Dist: tqdm (<5.0,>=4.66.2)
Requires-Dist: trl (<1.0,>=0.9.3)
Requires-Dist: peft (<0.14,>=0.8.0)
Requires-Dist: protobuf (<6.0.0,>=5.28.0)
Requires-Dist: datasets (<3.0,>=2.15.0)
Requires-Dist: simpleeval (<1.0,>=0.9.13)
Requires-Dist: aim (<4.0,>=3.19.0); extra == "aim"
Requires-Dist: wheel (<1.0,>=0.42.0); extra == "dev"
Requires-Dist: packaging (<25,>=23.2); extra == "dev"
Requires-Dist: ninja (<2.0,>=1.11.1.1); extra == "dev"
Requires-Dist: scikit-learn (<2.0,>=1.0); extra == "dev"
Requires-Dist: boto3 (<2.0,>=1.34); extra == "dev"
Requires-Dist: flash-attn (<3.0,>=2.5.3); extra == "flash-attn"
Requires-Dist: fms-acceleration (>=0.1); extra == "fms-accel"
Requires-Dist: auto-gptq (>0.4.2); extra == "gptq-dev"
Requires-Dist: optimum (>=1.15.0); extra == "gptq-dev"
Provides-Extra: aim
Provides-Extra: dev
Provides-Extra: flash-attn
Provides-Extra: fms-accel
Provides-Extra: gptq-dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 39653 characters]

WHEEL

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

RECORD

Path Digest Size
tuning/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/_version.py sha256=O82e8yUjRHAyzpOxLpXpGJZbuf3VENU0viCiiMxqvgI 411
tuning/sft_trainer.py sha256=H9EXmFbxj1KNxUg5glnGPmADlNW9ZL0vv7OD8LiXjAU 28007
tuning/config/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/config/configs.py sha256=NWHVkp53ubFyWOzePOMFJBszqDmdpIdTuUwdepHJ6Tw 6092
tuning/config/peft_config.py sha256=dyycs4h6asw0E9QHK1Q9j3pwwgtUyKiNL-ZveoF91-8 2989
tuning/config/tracker_configs.py sha256=9xLs_2hw4bTYIbmKPK-RIgJI0T5uwuF2tFVy1tIeJB8 2490
tuning/config/acceleration_configs/__init__.py sha256=6oUhiIjH2QWk3Tctq16FRHht20CDoeDMQz_elEc-ahQ 863
tuning/config/acceleration_configs/acceleration_framework_config.py sha256=TlJEwCCYM8PzGyEi3-RbEMyjxWa0hiUv9lap0hUBtIU 11677
tuning/config/acceleration_configs/attention_and_distributed_packing.py sha256=D3PjO8F3A_MtkAvZlb6-TRR9AH1rB2TOdwekHAeFEkQ 703
tuning/config/acceleration_configs/fused_ops_and_kernels.py sha256=gjqbz88S6muHfY9Vc6Kf9IL87ByQD_drL3_9KUUEmJg 2058
tuning/config/acceleration_configs/quantized_lora_config.py sha256=gGtlle6IaCphE0cL_jWXfkbcRvZxr939NGqGdtS5oU0 2093
tuning/config/acceleration_configs/utils.py sha256=JmIoBKiBtZBqiJbzvU5biPLqDhFmYH2xnmpP-60wXXQ 2842
tuning/trackers/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/trackers/aimstack_tracker.py sha256=G4lpFEqifHtexyAMvitOM-F_4hQePQ3WnRcj8I5qwn4 7416
tuning/trackers/filelogging_tracker.py sha256=Jmauno3Cq0tuXTk47yN252SV4Mc_PX3IK_vd0bOqdGU 3800
tuning/trackers/tracker.py sha256=EBbUGYYHtp5qpxhb7eBe7rIogJw4m6hWCxDrEOkqVvA 1246
tuning/trackers/tracker_factory.py sha256=ZJlOl6cJBmEktrtuI0-Fh--73YUu0Ec7kHvTNgq9p6Y 5327
tuning/trainercontroller/__init__.py sha256=il2PauyS3A93ZOcpniJi4FtMFoEeZTYsDB7p5bwF_kU 722
tuning/trainercontroller/callback.py sha256=noGNudz6VBmzNzywLcpl7tTpcBR_KlEFjUp74Q8mVlc 25798
tuning/trainercontroller/control.py sha256=MhvDBh3-APxLUFDGH2JS3dPImK8XfVhqT6z8DI44Hvg 1512
tuning/trainercontroller/patience.py sha256=VrlVnFFReWbSX665xlmGWTimFnz2-VZW7UEfuU7HFxw 2835
tuning/trainercontroller/controllermetrics/__init__.py sha256=jNNcviuQ2jZAp77KIoMgIP2pr7uP_LjrucN-gpkPjE4 1357
tuning/trainercontroller/controllermetrics/eval_metrics.py sha256=hm9eewEpzTRpuME7cEMqfc-rg42OeIAU5nZM-LCUE60 2178
tuning/trainercontroller/controllermetrics/history_based_metrics.py sha256=jQyQz1Cmgc4e8w1hb7TR2dR2RdL7n4-Iu6gAOQSIF-g 4902
tuning/trainercontroller/controllermetrics/loss.py sha256=zNLXXcuImviUermut1U3a_H3ZL3_5iF2YiNQzTNOBFk 2017
tuning/trainercontroller/controllermetrics/metrics.yaml sha256=LldSit3YLhdwus8LzdAIcN-PpWRdaC3Y0DM_hDfCDH8 193
tuning/trainercontroller/controllermetrics/metricshandler.py sha256=Z7BzwHf860MgRdIyQYFCQRPCXHryDcVeB4nXVF3c6qY 2602
tuning/trainercontroller/controllermetrics/per_process_state.py sha256=22YbGQGD1G2r-b4KcK-PgUOGmmT7Zi3J0JFOdCYBCE4 2339
tuning/trainercontroller/controllermetrics/trainingstate.py sha256=_AQEWRjFTAgksGvSb8lP7VAu4Z_ncpCZuGq0FpwMEN4 2185
tuning/trainercontroller/operations/__init__.py sha256=_CEDTGtncG_uB1WubjISJEpC0cMg8zS6prwzVlnV_W8 504
tuning/trainercontroller/operations/hfcontrols.py sha256=l7YS258r2zncioWTx_OW-4NPlt51qcMjeS45w4xJkJw 1658
tuning/trainercontroller/operations/logcontrol.py sha256=yoaMWJlN0uaS__oFuCaUCswZBdVAUa_re2y38GlMGJk 1770
tuning/trainercontroller/operations/operation.py sha256=ulP2CfHRozkKwCHbIjGU580WzRt1PzQODLM5O26_r9M 2409
tuning/utils/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/utils/config_utils.py sha256=Lath3sWn-qFPtAE4two-9aiYhPjkQlJ1DrpteHW3HkE 4588
tuning/utils/data_type_utils.py sha256=0L-JI8D9G2e6xW_xCWMMBhWntRNBh5tb83_9iERDl74 1940
tuning/utils/data_utils.py sha256=67NbBd7SLsTCEe9YqKHcE2NrWPsozpOdD6rDThMXyDc 1533
tuning/utils/error_logging.py sha256=muoVSrUTDxkVaQGfx4dxyed7kEDNsN_9x7LfLY_3GgA 1533
tuning/utils/evaluator.py sha256=y8STsjF7YbzGJjDA3DAP55-yCSUUcxvO3Dmp6glZGdw 3340
tuning/utils/import_utils.py sha256=Ps7cCltme-Hs6qxrxoKWypGP_xVjcqocBZWm7oNY6Gc 1119
tuning/utils/logging.py sha256=L7BiMsqCJjBkyXoRLUQrehTduF_jtTuxMxAi0zUtQng 2151
tuning/utils/merge_model_utils.py sha256=eJaJyoEZzapRJLKszWNK--ftdlNDiBbk2qMXpQj5X9w 8184
tuning/utils/preprocessing_utils.py sha256=RvQCI29GQeb9gP491mFgw9yyRyU22j1VZqTo1MXI3Wk 17117
tuning/utils/tokenizer_data_utils.py sha256=95KGgbaqwOtvLS44FHPh4uv-cDIbiZPYivYgWtk__00 2108
fms_hf_tuning-2.1.1.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
fms_hf_tuning-2.1.1.dist-info/METADATA sha256=ghnptjVhfqQf177a2INvgIWDOBKo8ua6FYkaicQfQKc 41635
fms_hf_tuning-2.1.1.dist-info/WHEEL sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc 91
fms_hf_tuning-2.1.1.dist-info/top_level.txt sha256=Rgm2UOuNF6xse8PiUfxdkDkdn2MTXlllKdkIrZeYz2s 7
fms_hf_tuning-2.1.1.dist-info/RECORD — —

top_level.txt

tuning