From 2129c64aede6818bb7df8221af4acf05c8df3322 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Fri, 2 Feb 2024 22:42:14 -0800
Subject: [PATCH] Update safe.py

---
 modules/safe.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/modules/safe.py b/modules/safe.py
index 608cb7c3..ff5a4ca4 100644
--- a/modules/safe.py
+++ b/modules/safe.py
@@ -1,3 +1,4 @@
+import torch
 import contextlib
 
 TypedStorage = None
@@ -35,5 +36,5 @@ def Extra(*args, **kwargs):
     return contextlib.nullcontext()
 
 
-unsafe_torch_load = None
+unsafe_torch_load = torch.load
 global_extra_handler = None