# automatically generated by g-sorcery
# please do not edit this file

EAPI=8

REALNAME="${PN}"
LITERALNAME="${PN}"
REALVERSION="${PV}"
DIGEST_SOURCES="yes"
PYTHON_COMPAT=( python{3_11,3_12,3_13,3_14} )
DISTUTILS_USE_PEP517=wheel

inherit python-r1 gs-pypi

DESCRIPTION="Run AI models smashed by Pruna AI [wheel]"

HOMEPAGE=""
LICENSE="All Rights Reserved"
SRC_URI="https://files.pythonhosted.org/packages/py310/${REALNAME::1}/${REALNAME}/${REALNAME//-/_}-${REALVERSION}-py310-none-any.whl"
SOURCEFILE="${REALNAME//-/_}-${REALVERSION}-py310-none-any.whl"
RESTRICT="test"

SLOT="0"
KEYWORDS="~amd64 ~x86"

IUSE="gpu"
DEPENDENCIES="dev-python/packaging[${PYTHON_USEDEP}]
	dev-python/torch[${PYTHON_USEDEP}]
	dev-python/onnx[${PYTHON_USEDEP}]
	>=dev-python/pillow-9.5.0[${PYTHON_USEDEP}]
	>=dev-python/scipy-1.10.1[${PYTHON_USEDEP}]
	>=dev-python/requests-2.31.0[${PYTHON_USEDEP}]
	dev-python/transformers[${PYTHON_USEDEP}]
	>=dev-python/pycryptodome-3.18.0[${PYTHON_USEDEP}]
	dev-python/speedster[${PYTHON_USEDEP}]
	dev-python/nebullvm[${PYTHON_USEDEP}]
	dev-python/diffusers[${PYTHON_USEDEP}]
	>=dev-python/numpy-1.23.5[${PYTHON_USEDEP}]
	<=dev-python/protobuf-3.20.2[${PYTHON_USEDEP}]
	dev-python/peft[${PYTHON_USEDEP}]
	dev-python/accelerate[${PYTHON_USEDEP}]
	dev-python/flatbuffers[${PYTHON_USEDEP}]
	dev-python/openvino-dev[${PYTHON_USEDEP}]
	dev-python/ctranslate2[${PYTHON_USEDEP}]
	dev-python/whisper-s2t[${PYTHON_USEDEP}]
	dev-python/dill[${PYTHON_USEDEP}]
	dev-python/platformdirs[${PYTHON_USEDEP}]
	gpu? ( dev-python/tensorrt[${PYTHON_USEDEP}] )
	gpu? ( dev-python/onnx-graphsurgeon[${PYTHON_USEDEP}] )
	gpu? ( dev-python/nvtx[${PYTHON_USEDEP}] )
	gpu? ( dev-python/flash-attn[${PYTHON_USEDEP}] )
	gpu? ( dev-python/stable-fast[${PYTHON_USEDEP}] )
	gpu? ( dev-python/xformers[${PYTHON_USEDEP}] )
	gpu? ( dev-python/polygraphy[${PYTHON_USEDEP}] )
	gpu? ( dev-python/DeepCache[${PYTHON_USEDEP}] )
	gpu? ( dev-python/auto-gptq[${PYTHON_USEDEP}] )
	gpu? ( dev-python/cuda-python[${PYTHON_USEDEP}] )
	virtual/allow-pypi-wheels"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"