diff --git a/requirements.txt b/requirements.txt index b9d52c5b2..aca426bb2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,8 +10,9 @@ pytest-benchmark requests tabulate git+https://github.com/huggingface/pytorch-image-models.git@730b907 -# this version of transformers is required as per this page https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1 -transformers==4.38.1 +# this version of transformers is required by linger-kernel +# https://github.com/linkedin/Liger-Kernel/blob/main/pyproject.toml#L23 +transformers==4.44.2 MonkeyType psutil pyyaml