Update birefnet.py

birefnet.py

@@ -1993,9 +1993,22 @@ def patches2image(patches, grid_h=2, grid_w=2, patch_ref=None, transformation='(
     return image
 
 class BiRefNet(
-    PreTrainedModel
-):
+     PreTrainedModel
+ ):
     config_class = BiRefNetConfig
+
+    
+    @property
+    def all_tied_weights_keys(self):
+        keys = getattr(self, "_tied_weights_keys", None)
+        if keys is None:
+            return {}
+        if isinstance(keys, dict):
+            return keys
+        try:
+            return {k: None for k in keys}
+        except TypeError:
+            return {}
     def __init__(self, bb_pretrained=True, config=BiRefNetConfig()):
         super(BiRefNet, self).__init__(config)
         bb_pretrained = config.bb_pretrained