Upload folder using huggingface_hub

modeling_flmr.py

@@ -645,7 +645,10 @@ class FLMRModelForRetrieval(FLMRPretrainedModelForRetrieval):
             self._tied_weights_keys += ["context_vision_encoder", "context_vision_projection"]
 
         if self.config.load_cpu_extension:
-            FLMRModelForRetrieval.try_load_torch_extensions()
+            try:
+                FLMRModelForRetrieval.try_load_torch_extensions()
+            except Exception as e:
+                raise(f"Unable to load `segmented_maxsim.cpp`. hf-hub does not download this file automatically. Please download it manually from `https://huggingface.co/LinWeizheDragon/PreFLMR_ViT-L/blob/main/segmented_maxsim.cpp` and put it under the same folder as the model file.\n {e}")
 
         if self.config.mask_punctuation:
             self.skiplist = {