Update app.py

app.py

@@ -22,8 +22,12 @@ _original_adjust_tied = PreTrainedModel._adjust_tied_keys_with_tied_pointers
 
 
 def _safe_adjust_tied_keys_with_tied_pointers(self, tied_weights_keys_by_pointers):
-    if not hasattr(self, "all_tied_weights_keys"):
-        self.all_tied_weights_keys = set(getattr(self, "_tied_weights_keys", []) or [])
+    existing = getattr(self, "all_tied_weights_keys", None)
+    if existing is None:
+        tied = getattr(self, "_tied_weights_keys", []) or []
+        self.all_tied_weights_keys = {k: True for k in tied}
+    elif not hasattr(existing, "keys"):
+        self.all_tied_weights_keys = {k: True for k in existing}
     return _original_adjust_tied(self, tied_weights_keys_by_pointers)