From 790ceb254a94fd1d22b100cff1b0ac949215dff6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Quentin=20Gallou=C3=A9dec?=
 <45557362+qgallouedec@users.noreply.github.com>
Date: Wed, 3 Jul 2024 19:35:20 +0200
Subject: [PATCH] `safetensors[torch]`

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 4ecbb42e66..29c831ee9f 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,7 @@ def get_version() -> str:
 
 extras["torch"] = [
     "torch",
-    "safetensors",
+    "safetensors[torch]",
 ]
 extras["hf_transfer"] = [
     "hf_transfer>=0.1.4",  # Pin for progress bars