Commit

app.py

@@ -10,6 +10,7 @@ from transformers import (
 from peft import LoraConfig, get_peft_model
 from supabase import create_client
 from sentence_transformers import SentenceTransformer
+from safetensors.torch import load_file as safe_load
 
 # ── Supabase setup ─────────────────────────────────────────────────────────
 SUPA_URL = os.getenv("SUPABASE_URL")
@@ -94,10 +95,8 @@ def load_generator():
     model = get_peft_model(base, peft_config)
 
     # 5) Manually load adapter weights
-    adapter_path = os.path.join(
-        LOCAL_REPO, "adapter_model", "pytorch_model.bin"
-    )
-    state_dict = torch.load(adapter_path, map_location="cpu")
+    adapter_path = os.path.join(LOCAL_REPO, "adapter_model.safetensors")
+    state_dict = safe_load(adapter_path)
     model.load_state_dict(state_dict, strict=False)
     model.eval()