diff --git a/setup.py b/setup.py index 28f71f789..5abd08fc6 100644 --- a/setup.py +++ b/setup.py @@ -153,6 +153,7 @@ extras_require = { "llmcompressor": [ "llmcompressor==0.5.1", ], + "mistral": ["mistral-common==1.5.6"], } install_requires, dependency_links, extras_require_build = parse_requirements(