fms-hf-tuning

View on PyPIReverse Dependencies (0)

2.3.0 fms_hf_tuning-2.3.0-py3-none-any.whl

Wheel Details

Project: fms-hf-tuning
Version: 2.3.0
Filename: fms_hf_tuning-2.3.0-py3-none-any.whl
Download: [link]
Size: 98256
MD5: 12a8c010766122d3cbb209c1355de643
SHA256: 3b8901a4f097f5c3009855eb11c6037facd3ce3014841c838e0e38b64c6d3343
Uploaded: 2024-12-19 21:15:49 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: fms-hf-tuning
Version: 2.3.0
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.45)
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 (<0.12,>=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: 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: aim (<4.0,>=3.19.0); extra == "aim"
Requires-Dist: mlflow; extra == "mlflow"
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: dev
Provides-Extra: flash-attn
Provides-Extra: aim
Provides-Extra: mlflow
Provides-Extra: fms-accel
Provides-Extra: gptq-dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 42251 characters]

WHEEL

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

RECORD

Path Digest Size
tuning/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/_version.py sha256=ChsIHG8bRc-eXUbXOgv4Fm4DstSKLq9FpsTAsaMeR08 411
tuning/sft_trainer.py sha256=LzLTfdkFRInthtLaGvL6rFThuRMlIw3fMXG84lvaHA0 28033
tuning/config/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/config/configs.py sha256=b_s0czrn2BOxHXD4Bnb_1n-3uQ48Fcrrdm7nxcOTjUg 7166
tuning/config/peft_config.py sha256=dyycs4h6asw0E9QHK1Q9j3pwwgtUyKiNL-ZveoF91-8 2989
tuning/config/tracker_configs.py sha256=vtiaYza289qxJ5dbHeBevKBwuQJeqPiVtEpqONLdUdw 3130
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/data/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/data/data_config.py sha256=NQbHvkN74vDXW54Z_mP6MMmE5lU3N_g0po3iEcRRs1w 5736
tuning/data/data_handlers.py sha256=aqFpgvIZTC7kVlQCMcTIilzLgriUzeN6yqKxyE4V4_M 5348
tuning/data/data_preprocessing_utils.py sha256=NYabvH9hifW5Y7XA-G5mWhNaO9M0jZHv59tAfUASBVg 3267
tuning/data/data_processors.py sha256=MbGKAN05ixgeUZvGBsqeb15EiTIDsW0HKPjgOIyC9aI 13674
tuning/data/setup_dataprocessor.py sha256=yB4R4ICm4TZ4-6BcGvux6raDLM5hengApEjAJfU4mVM 13909
tuning/trackers/__init__.py sha256=mac32u5CDFCwfgvWJ9hhlFP2bIjPnpZMhoQwRpFJp6w 584
tuning/trackers/aimstack_tracker.py sha256=HlKJcSoSh7AzvSQHjPG9_6K8j3utMgwO_8B3BnLHls4 7424
tuning/trackers/filelogging_tracker.py sha256=Jmauno3Cq0tuXTk47yN252SV4Mc_PX3IK_vd0bOqdGU 3800
tuning/trackers/mlflow_tracker.py sha256=Id3Xj555RZMDBCD1-mu0DDWgO4YkSng6LwYhnoxO9fk 7861
tuning/trackers/tracker.py sha256=EBbUGYYHtp5qpxhb7eBe7rIogJw4m6hWCxDrEOkqVvA 1246
tuning/trackers/tracker_factory.py sha256=HA7ZXw4nt0rS18uWUiBKBEwk9IuvrNqZiBsX2B17zus 6600
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/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/tokenizer_data_utils.py sha256=95KGgbaqwOtvLS44FHPh4uv-cDIbiZPYivYgWtk__00 2108
tuning/utils/utils.py sha256=R94Lx6i272yprQatJ2WBDZ1DPMYKKggRSmLVnGZ44YE 2828
fms_hf_tuning-2.3.0.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
fms_hf_tuning-2.3.0.dist-info/METADATA sha256=f-gdUHEXHdv3QaNuZJqDWKpd6aUYVokWnCjtfXG4rLI 44426
fms_hf_tuning-2.3.0.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
fms_hf_tuning-2.3.0.dist-info/top_level.txt sha256=Rgm2UOuNF6xse8PiUfxdkDkdn2MTXlllKdkIrZeYz2s 7
fms_hf_tuning-2.3.0.dist-info/RECORD

top_level.txt

tuning