update m3 FT

Author: 545999961

FlagEmbedding/finetune/embedder/encoder_only/m3/runner.py

@@ -38,8 +38,8 @@ def get_model(
         colbert_dim: int = -1,
         cache_dir: str = None
     ):
+        cache_folder = os.getenv('HF_HUB_CACHE', None) if cache_dir is None else cache_dir
         if not os.path.exists(model_name_or_path):
-            cache_folder = os.getenv('HF_HUB_CACHE', None) if cache_dir is None else cache_dir
             model_name_or_path = snapshot_download(
                 repo_id=model_name_or_path,
                 cache_dir=cache_folder,
@@ -48,6 +48,7 @@ def get_model(
 
         model = AutoModel.from_pretrained(
             model_name_or_path,
+            cache_dir=cache_folder,
             trust_remote_code=trust_remote_code
         )
         colbert_linear = torch.nn.Linear(