From 802929f5f2ad7a08e9df2fb908a1109921a14e44 Mon Sep 17 00:00:00 2001
From: somebody <onesomegit@outlook.com>
Date: Fri, 7 Jul 2023 14:54:40 -0500
Subject: [PATCH] Patch safetensors again

---
 modeling/lazy_loader.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modeling/lazy_loader.py b/modeling/lazy_loader.py
index 1d2a6829..da2ec989 100644
--- a/modeling/lazy_loader.py
+++ b/modeling/lazy_loader.py
@@ -520,8 +520,8 @@ def use_lazy_load(
 
         torch.load = torch_load
 
-        #if HAS_SAFETENSORS:
-            #patch_safetensors(callback)
+        if HAS_SAFETENSORS:
+            patch_safetensors(callback)
 
         if dematerialized_modules:
             # Most devices can just use Accelerate's implementation, but the Transformers on