# 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} )
DISTUTILS_USE_PEP517=standalone

inherit python-r1 gs-pypi

DESCRIPTION="An efficient, flexible and full-featured toolkit for fine-tuning large models"

HOMEPAGE="https://github.com/InternLM/xtuner"
LICENSE="Apache-2.0"
SRC_URI="https://files.pythonhosted.org/packages/source/${REALNAME::1}/${REALNAME}/${REALNAME}-${REALVERSION}.tar.gz"
SOURCEFILE="${REALNAME}-${REALVERSION}.tar.gz"
RESTRICT="test"

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

IUSE="all deepspeed"
DEPENDENCIES="dev-python/bitsandbytes[${PYTHON_USEDEP}]
	dev-python/datasets[${PYTHON_USEDEP}]
	dev-python/einops[${PYTHON_USEDEP}]
	dev-python/loguru[${PYTHON_USEDEP}]
	dev-python/mmengine[${PYTHON_USEDEP}]
	dev-python/openpyxl[${PYTHON_USEDEP}]
	dev-python/peft[${PYTHON_USEDEP}]
	dev-python/scikit-image[${PYTHON_USEDEP}]
	dev-python/scipy[${PYTHON_USEDEP}]
	dev-python/sentencepiece[${PYTHON_USEDEP}]
	dev-python/tiktoken[${PYTHON_USEDEP}]
	dev-python/torch[${PYTHON_USEDEP}]
	dev-python/torchvision[${PYTHON_USEDEP}]
	dev-python/transformers[${PYTHON_USEDEP}]
	dev-python/transformers-stream-generator[${PYTHON_USEDEP}]
	all? ( dev-python/bitsandbytes[${PYTHON_USEDEP}] )
	all? ( dev-python/datasets[${PYTHON_USEDEP}] )
	all? ( dev-python/einops[${PYTHON_USEDEP}] )
	all? ( dev-python/loguru[${PYTHON_USEDEP}] )
	all? ( dev-python/mmengine[${PYTHON_USEDEP}] )
	all? ( dev-python/openpyxl[${PYTHON_USEDEP}] )
	all? ( dev-python/peft[${PYTHON_USEDEP}] )
	all? ( dev-python/scikit-image[${PYTHON_USEDEP}] )
	all? ( dev-python/scipy[${PYTHON_USEDEP}] )
	all? ( dev-python/sentencepiece[${PYTHON_USEDEP}] )
	all? ( dev-python/tiktoken[${PYTHON_USEDEP}] )
	all? ( dev-python/torch[${PYTHON_USEDEP}] )
	all? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	all? ( dev-python/transformers[${PYTHON_USEDEP}] )
	all? ( dev-python/transformers-stream-generator[${PYTHON_USEDEP}] )
	all? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	all? ( dev-python/mpi4py-mpich[${PYTHON_USEDEP}] )
	all? ( dev-python/modelscope[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/bitsandbytes[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/datasets[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/einops[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/loguru[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/mmengine[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/openpyxl[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/peft[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/scikit-image[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/scipy[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/sentencepiece[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/tiktoken[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/torch[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/transformers[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/transformers-stream-generator[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/mpi4py-mpich[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"