From 59c96b5b7aea2eaf4ad8bab70794c8f3d41edccf Mon Sep 17 00:00:00 2001 From: Henk Date: Mon, 15 May 2023 22:38:12 +0200 Subject: [PATCH] Unban fix --- modeling/lazy_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modeling/lazy_loader.py b/modeling/lazy_loader.py index a948821e..5a27d549 100644 --- a/modeling/lazy_loader.py +++ b/modeling/lazy_loader.py @@ -196,8 +196,8 @@ class RestrictedUnpickler(pickle.Unpickler): return collections.OrderedDict elif module == "torch._utils" and name == "_rebuild_tensor_v2": return torch._utils._rebuild_tensor_v2 - elif module == "torch._tensor" and name == "_rebuild_tensor_v2": - return torch._tensor._rebuild_tensor_v2 + elif module == "torch._tensor" and name == "_rebuild_from_type_v2": + return torch._tensor._rebuild_from_type_v2 elif module == "torch" and name in ( "DoubleStorage", "FloatStorage",