rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Unban torch._tensor._rebuild_tensor_v2

Browse Source
This commit is contained in:
Henk
2023-05-15 21:44:01 +02:00
parent 205c64f1ea
commit 56443bc7ea
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modeling/lazy_loader.py
View File

@@ -196,6 +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" and name in (
"DoubleStorage",
"FloatStorage",