Update utils/models.py

utils/models.py

@@ -21,6 +21,18 @@ except Exception:
     pass
 collection = chroma_client.create_collection(name=config.collection_name)
 
+# --- THE MONKEY PATCH ---
+# Intercept the breaking change in Transformers v4.45+ for custom models
+_orig_getattr = torch.nn.Module.__getattr__
+
+def _patched_getattr(self, name):
+    if name == "all_tied_weights_keys":
+        return {}
+    return _orig_getattr(self, name)
+
+torch.nn.Module.__getattr__ = _patched_getattr
+# ------------------------
+
 # 3. Load VLM
 logger.info(f"Loading VLM ({config.vlm_model_id})...")
 vlm_model = AutoModelForCausalLM.from_pretrained(