add hf ckpt

models/farslip_model.py

@@ -6,6 +6,7 @@ import pandas as pd
 import pyarrow.parquet as pq
 import torch.nn.functional as F
 from PIL import Image
+from huggingface_hub import hf_hub_download
 
 try:
     # FarSLIP must use its vendored open_clip fork (checkpoint format / model defs differ).
@@ -27,6 +28,8 @@ class FarSLIPModel:
         
         self.device = device if device else ("cuda" if torch.cuda.is_available() else "cpu")
         self.model_name = model_name
+        if 'hf' in ckpt_path:
+            ckpt_path = hf_hub_download("ZhenShiL/FarSLIP", "FarSLIP2_ViT-B-16.pt")
         self.ckpt_path = ckpt_path
         self.embedding_path = embedding_path