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 fix

Browse Source
This commit is contained in:
Henk
2023-05-15 22:38:12 +02:00
parent c5100b4eab
commit 59c96b5b7a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modeling/lazy_loader.py
View File

@@ -196,8 +196,8 @@ class RestrictedUnpickler(pickle.Unpickler):
return collections.OrderedDict return collections.OrderedDict
elif module == "torch._utils" and name == "_rebuild_tensor_v2": elif module == "torch._utils" and name == "_rebuild_tensor_v2":
return torch._utils._rebuild_tensor_v2 return torch._utils._rebuild_tensor_v2
elif module == "torch._tensor" and name == "_rebuild_tensor_v2": elif module == "torch._tensor" and name == "_rebuild_from_type_v2":
return torch._tensor._rebuild_tensor_v2 return torch._tensor._rebuild_from_type_v2
elif module == "torch" and name in ( elif module == "torch" and name in (
"DoubleStorage", "DoubleStorage",
"FloatStorage", "FloatStorage",