From 332c0d03496230c7cd74ceb00032cb2a791a0bd3 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 2 Feb 2024 21:12:00 -0800 Subject: [PATCH] Update safe.py --- modules/safe.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/safe.py b/modules/safe.py index 068d6083..608cb7c3 100644 --- a/modules/safe.py +++ b/modules/safe.py @@ -1,3 +1,5 @@ +import contextlib + TypedStorage = None @@ -29,8 +31,8 @@ def load_with_extra(filename, extra_handler=None, *args, **kwargs): pass -class Extra: - pass +def Extra(*args, **kwargs): + return contextlib.nullcontext() unsafe_torch_load = None