fms-model-optimizer

View on PyPIReverse Dependencies (0)

0.2.0 fms_model_optimizer-0.2.0-py3-none-any.whl

Wheel Details

Project: fms-model-optimizer
Version: 0.2.0
Filename: fms_model_optimizer-0.2.0-py3-none-any.whl
Download: [link]
Size: 213015
MD5: a28ac73055ce0f360e66821fb4b6439b
SHA256: 2c259894744bbb7e7946a0c986cf7fbe536c3cf9ae90965d3e3953e519bf9b8e
Uploaded: 2024-12-13 17:51:40 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: fms-model-optimizer
Version: 0.2.0
Summary: Quantization Techniques
Project-Url: homepage, https://github.com/foundation-model-stack/fms-model-optimizer
Project-Url: source, https://github.com/foundation-model-stack/fms-model-optimizer
Project-Url: issues, https://github.com/foundation-model-stack/fms-model-optimizer/issues
License: Apache-2.0
Classifier: Development Status :: 3 - Alpha
Classifier: License :: OSI Approved :: Apache Software License
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: POSIX :: Linux
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: Implementation :: CPython
Requires-Python: <3.12,>=3.9
Requires-Dist: torch (<2.5,>=2.1)
Requires-Dist: numpy
Requires-Dist: tqdm
Requires-Dist: tensorboard
Requires-Dist: notebook
Requires-Dist: torchvision (>=0.8)
Requires-Dist: transformers
Requires-Dist: accelerate
Requires-Dist: evaluate
Requires-Dist: datasets
Requires-Dist: huggingface_hub
Requires-Dist: pandas
Requires-Dist: safetensors
Requires-Dist: ninja
Requires-Dist: pre-commit (<5.0,>=3.0.4); extra == "dev"
Requires-Dist: llmcompressor; extra == "fp8"
Requires-Dist: auto_gptq; extra == "gptq"
Requires-Dist: matplotlib; extra == "visualize"
Requires-Dist: graphviz; extra == "visualize"
Requires-Dist: pygraphviz; extra == "visualize"
Provides-Extra: dev
Provides-Extra: fp8
Provides-Extra: gptq
Provides-Extra: visualize
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4634 characters]

WHEEL

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

RECORD

Path Digest Size
fms_mo/__init__.py sha256=1Ndb0FBdKf1r0o5Y2k7wsTL8OPP5gcCnETnfbGtjZ_A 1077
fms_mo/_version.py sha256=H-qsvrxCpdhaQzyddR-yajEqI71hPxLa4KxzpP3uS1g 411
fms_mo/calib.py sha256=Mapz0vjknq8ax7bETYmpXHMXdI9dp23LmkjFQzuDlKE 27653
fms_mo/dq.py sha256=zrGYjHBYTQZt8Il4o14qvH0MyaTdg3Un6NGU_yRv9Ic 9005
fms_mo/prep.py sha256=emuiRYA36flkVRaX2kYFK8HWk4fLWhdEq7crkeWTmjw 38098
fms_mo/run_quant.py sha256=rwePY3FFKCnWmmu_JAap3i5zf9jMy-Q6Dciqv2D1NbY 9113
fms_mo/training_args.py sha256=k9NO4r64KLt99jiRylPDzE6AYkYiDlJm93Ip9qoFrpE 5085
fms_mo/custom_ext_kernels/__init__.py sha256=49VbzI3jcbYu8Af39_4bweQy2Fi0mCUC9NAGU9wgVIk 590
fms_mo/custom_ext_kernels/cutlass_i8i32gemm.cpp sha256=KYRppZZI1dkklvM7aiwdMxJQ5rpA7WkIQlXJ3esjw3I 2431
fms_mo/custom_ext_kernels/cutlass_i8i32gemm.cu sha256=SKjMR1iceFeQXCnHegg7QuTobhT1FkoUr5cnoojXcrU 12499
fms_mo/custom_ext_kernels/ext_kernel_cublas.cpp sha256=gcGhwvTUfxVzhh7EgnB0OP9wOWViZ5RmjPVkfEx4qEU 18914
fms_mo/custom_ext_kernels/utils.py sha256=deuoYpYs8_1OB7K0S35Ofn6VFUeSRb6WpQg0e3WOWls 42916
fms_mo/fx/dynamo_utils.py sha256=52i8l40dKF6t8V9NDf5kfeNLdd5KNBuqstgHQnZg6R0 47168
fms_mo/fx/utils.py sha256=arNfpyjLBY5lT3RPYVwzJCPptSnKXh8TqzhdZgcAoII 22672
fms_mo/modules/__init__.py sha256=nPjLzdnx7jBsQhYImKTkYI1vnSeftGK1wy13iPqI1z8 993
fms_mo/modules/bmm.py sha256=bPrzpdTzr2Nufo8CzfUAIZOEo47m8fgJrYmxsahz1Ec 18700
fms_mo/modules/conv.py sha256=cJ48zA22gqU83c22gUjsyexGIhDKD3Kk5ofpWVGzadI 61704
fms_mo/modules/linear.py sha256=mcuztO2KtlB5DIf50d8MQmSBvjul-pW89mHb1hhJSV0 67380
fms_mo/modules/lstm.py sha256=IWfRXAO4y9Dy7H2_sYmKGZVxNUbWtBEtcoD0qTYZf-c 23794
fms_mo/quant/README.md sha256=Z7R5AEfHsJ3266V0zvRb7GlnHZ4pUQB5KdU4JqKjY7s 1828
fms_mo/quant/ptq.py sha256=4TmgmsQWZRmaqcybm-7x0eHIGTCcQXauO8ZiawwIJhQ 105372
fms_mo/quant/quantizers.py sha256=dM_k57WXY4ir5OqEhSIRjTR09iOcwoq4DxPEB8JQlUo 202210
fms_mo/recipes/dq.json sha256=MgYb4Nnk8bC6JMp6rwqmnO9nZU2sOTLSHGCrl2fAcSU 317
fms_mo/recipes/ptq_int8.json sha256=bkkgCHyQOvkCbBPXEHx4Pc3flPVm1c8QOt9PBTYVEvI 980
fms_mo/recipes/qat_int8.json sha256=VZnWr4NkMUQfY2q7xzMfNj-sdC9SKYsMP_bFa0UZAQU 576
fms_mo/utils/README.md sha256=td6bLib918PMYot0LZjiJBq7o3C2H3u1SJcFIm_8W34 11618
fms_mo/utils/__init__.py sha256=MYgXho7ZsXJvhDaCVTs3O8rNXN97Aap6eynYaTDIKsc 623
fms_mo/utils/calib_data.py sha256=FQMtGJvMFAS-uTxELrhTTganebMBgMS4PJDQCX_z3Xc 13658
fms_mo/utils/custom_gptq_models.py sha256=AH18AGScKgL96I9qHk8m8fac5Q5Ahsc4_1mvSjknWA0 1997
fms_mo/utils/dq_utils.py sha256=dgMOH6LrSoTThhtkxhcfRipEsBZcuxbUDMWkqe65Lic 4657
fms_mo/utils/eval_utils.py sha256=urU2JeB4GuW84-oAMfv-RVrFW1h4bXoZ8bGngw9wtno 5988
fms_mo/utils/import_utils.py sha256=_WDV1YvLRs6LnaA8FhIzPxLgRGXyu0NL64mqjn6HxfM 1224
fms_mo/utils/qconfig_utils.py sha256=nivxIq86WDdg7TtwAsJuIIp-0ymEvktEbCwqtnW5zCQ 30172
fms_mo/utils/torchscript_utils.py sha256=NAuA3ysEjAx7hQ2ZMa38b1cXoStL-dRnmKf5uAUrogc 70546
fms_mo/utils/utils.py sha256=3gYP5T9uT_yAa3YWht6HBol3FVHtTeY_mpcwGske3Lc 13620
fms_model_optimizer-0.2.0.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
fms_model_optimizer-0.2.0.dist-info/METADATA sha256=e9l4QpMNt7WAJOPd7lfg9ZUSU_OEmQQhsSr2m2_DiKs 6382
fms_model_optimizer-0.2.0.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
fms_model_optimizer-0.2.0.dist-info/top_level.txt sha256=H4h6eH-UiQGCPCqJOaStyjKfx-yi-mmOtqbw0fuLZic 7
fms_model_optimizer-0.2.0.dist-info/RECORD

top_level.txt

fms_mo