out cache in biencode

bi_encoder.py

@@ -4,8 +4,8 @@ import torch
 from transformers import AutoTokenizer, AutoModel
 from sklearn.metrics.pairwise import cosine_similarity
 
-tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased", cache_dir="/model/transformers_cache")
-bert_model = AutoModel.from_pretrained("distilbert-base-uncased", cache_dir="/model/transformers_cache")
+tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
+bert_model = AutoModel.from_pretrained("distilbert-base-uncased")
 
 
 def mean_pool(token_embeds: torch.tensor, attention_mask: torch.tensor) -> torch.tensor: