Update safe.py

This commit is contained in:
lllyasviel
2024-02-02 22:42:14 -08:00
parent 1edd626f7c
commit 2129c64aed

View File

@@ -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