Debug: log reliability_binary flag + reliability_head shape + ckpt info

app.py

@@ -104,6 +104,15 @@ COLLATER = InferenceCollater(
     prot_max_len=MODEL_ARGS.prot_max_len,
 )
 PROMPT = "Swiss-Prot description: "
+
+print(f"[debug] MODEL_ARGS.reliability_binary = {getattr(MODEL_ARGS, 'reliability_binary', '<missing>')}")
+print(f"[debug] BLIP2.reliability_binary     = {getattr(BLIP2, 'reliability_binary', '<missing>')}")
+_rh_w = BLIP2.reliability_head[1].weight
+print(f"[debug] reliability_head output dim  = {_rh_w.shape[0]} (binary=2, 4-class=4)")
+print(f"[debug] ckpt mtime / size            = {os.path.getmtime(CKPT_PATH):.0f}, {os.path.getsize(CKPT_PATH):,} bytes")
+_ckpt_keys_head = [k for k in _state_dict if "reliability_head" in k]
+for _k in _ckpt_keys_head:
+    print(f"[debug] ckpt has {_k}: shape={tuple(_state_dict[_k].shape)}")
 print("Ready.")