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