diff --git a/pyproject.toml b/pyproject.toml index 8a300f0ea8..60194b5f73 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,7 +37,7 @@ all = [ "tokenizers", # pythia, falcon, redpajama "requests", # litgpt.data "litdata", # litgpt.data - "litserve" # litgpt.deploy + "litserve", # litgpt.deploy "zstandard", # litgpt.data.prepare_slimpajama.py "pandas", # litgpt.data.prepare_starcoder.py "pyarrow", # litgpt.data.prepare_starcoder.py