From c74f603ea26fd398d879fd3532bbb42053c49e03 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Thu, 15 Aug 2024 00:23:31 -0700 Subject: [PATCH] remove super call --- backend/operations.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/operations.py b/backend/operations.py index 636e9df4..9dd3d5f3 100644 --- a/backend/operations.py +++ b/backend/operations.py @@ -350,7 +350,10 @@ class ForgeOperationsGGUF(ForgeOperations): self.bias = state_dict[prefix + 'bias'].to(device=self.dummy.device) del self.dummy else: - super()._load_from_state_dict(state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs) + if prefix + 'weight' in state_dict: + self.weight = state_dict[prefix + 'weight'] + if prefix + 'bias' in state_dict: + self.bias = state_dict[prefix + 'bias'] def _apply(self, fn, recurse=True): if self.weight is not None: