mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Last debug
This commit is contained in:
@@ -144,7 +144,6 @@ class LazyloadPatches:
|
||||
LazyloadPatches._load_state_dict_into_meta_model
|
||||
)
|
||||
torch.nn.Module._load_from_state_dict = LazyloadPatches._torch_load_from_state_dict
|
||||
# torch.nn.Module._load_from_state_dict = _agn
|
||||
|
||||
def __exit__(exc_type, exc_value, exc_traceback) -> None:
|
||||
transformers.modeling_utils._load_state_dict_into_meta_model = LazyloadPatches.old_load_state_dict
|
||||
|
Reference in New Issue
Block a user