diff --git a/requirements.txt b/requirements.txt index 3c01461..191d853 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,4 @@ transformers datasets accelerate tqdm -llm-compressor @ git+https://github.com/vllm-project/llm-compressor.git +llmcompressor @ git+https://github.com/vllm-project/llm-compressor.git diff --git a/setup.py b/setup.py index c2b015d..3dcd85f 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,7 @@ "datasets", "accelerate", "tqdm", - "llm-compressor @ git+https://github.com/vllm-project/llm-compressor.git" + "llmcompressor @ git+https://github.com/vllm-project/llm-compressor.git" ], classifiers=[ "Programming Language :: Python :: 3",