diff --git a/pyproject.toml b/pyproject.toml index c84af8b..59a03be 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -26,7 +26,7 @@ train = [ "pyarrow>=15.0.0", "tqdm>=4.66.1", "huggingface_hub>=0.30.0", - "safetensors>=0.4.5", + "safetensors>=0.7.0", "tokenizers>=0.20.0", ] serve = [