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 import contextlib
TypedStorage = None TypedStorage = None
@@ -35,5 +36,5 @@ def Extra(*args, **kwargs):
return contextlib.nullcontext() return contextlib.nullcontext()
unsafe_torch_load = None unsafe_torch_load = torch.load
global_extra_handler = None global_extra_handler = None