Metadata-Version: 2.1
Name: pruna
Version: 0.2.4
Summary: Smash your AI models
License: All Rights Reserved
Author: Pruna AI
Author-email: hello@pruna.ai
Requires-Python: >=3.9,<3.13
Classifier: License :: Other/Proprietary License
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 :: 3.12
Provides-Extra: autoawq
Provides-Extra: cpu
Provides-Extra: dev
Provides-Extra: full
Provides-Extra: gptq
Provides-Extra: stable-fast
Provides-Extra: tests
Requires-Dist: ConfigSpace (>=1.2.1)
Requires-Dist: DeepCache
Requires-Dist: autoawq ; extra == "autoawq" or extra == "full"
Requires-Dist: bitsandbytes
Requires-Dist: blobfile
Requires-Dist: build ; extra == "dev"
Requires-Dist: codecarbon
Requires-Dist: colorama
Requires-Dist: coverage ; extra == "tests"
Requires-Dist: ctranslate2 (==4.5.0)
Requires-Dist: cython
Requires-Dist: datasets (>=0.34)
Requires-Dist: diffusers (>=0.21.4)
Requires-Dist: docutils ; extra == "tests"
Requires-Dist: gptqmodel ; extra == "gptq"
Requires-Dist: hqq
Requires-Dist: huggingface-hub[hf-xet] (>=0.30.0)
Requires-Dist: ipywidgets (>=8.1.5)
Requires-Dist: jsonschema
Requires-Dist: jupyterlab ; extra == "dev" or extra == "tests"
Requires-Dist: lago-python-client
Requires-Dist: librosa
Requires-Dist: mypy ; extra == "tests"
Requires-Dist: notebook ; extra == "dev"
Requires-Dist: numpy (>=1.24.4)
Requires-Dist: numpydoc-validation ; extra == "dev" or extra == "tests"
Requires-Dist: opentelemetry-api (>=1.30.0)
Requires-Dist: opentelemetry-exporter-otlp (>=1.29.0)
Requires-Dist: opentelemetry-sdk (>=1.30.0)
Requires-Dist: openvino
Requires-Dist: optimum
Requires-Dist: optimum-quanto (>=0.2.5)
Requires-Dist: optuna
Requires-Dist: packaging
Requires-Dist: peft
Requires-Dist: pillow (>=9.5.0)
Requires-Dist: platformdirs
Requires-Dist: pre-commit ; extra == "dev"
Requires-Dist: protobuf
Requires-Dist: pyc-wheel ; extra == "dev"
Requires-Dist: pynvml
Requires-Dist: pytest (==7.4.4) ; extra == "tests"
Requires-Dist: pytest-cov ; extra == "tests"
Requires-Dist: python-dotenv
Requires-Dist: pytorch-lightning
Requires-Dist: requests (>=2.31.0)
Requires-Dist: ruff ; extra == "dev" or extra == "tests"
Requires-Dist: sentencepiece
Requires-Dist: setuptools
Requires-Dist: soundfile
Requires-Dist: stable-fast-pruna (==1.0.7) ; extra == "stable-fast" or extra == "full"
Requires-Dist: thop
Requires-Dist: timm
Requires-Dist: torch (==2.7.0)
Requires-Dist: torch_pruning
Requires-Dist: torchao
Requires-Dist: torchmetrics[image]
Requires-Dist: torchvision (==0.22.0)
Requires-Dist: transformers
Requires-Dist: trl
Requires-Dist: twine ; extra == "dev"
Requires-Dist: types-PyYAML ; extra == "tests"
Requires-Dist: wget
Requires-Dist: whisper-s2t (==1.3.0)
Requires-Dist: xformers (==0.0.30) ; extra == "stable-fast" or extra == "full"
