How to load bert_base-augmented-batch_size=128-lr=2e-5-max_gloss=6 model in offline mode in jupyter nb
I have downloaded "bert_base-augmented-batch_size=128-lr=2e-5-max_gloss=6" model locally but while trying to use it in jupyter notebook getting error:

```
HTTPError: 404 Client Error: Not Found for url: https://huggingface.co/api/models//bert_base-augmented-batch_size=128-lr=2e-5-max_gloss=6
```
How to fix this?
code I am using to use the model is as below:

```
import torch
import math
from transformers import BertModel, BertConfig, BertPreTrainedModel, BertTokenizer

class BertWSD(BertPreTrainedModel):
    def __init__(self, config):
        super().__init__(config)

        self.bert = BertModel(config)
        self.dropout = torch.nn.Dropout(config.hidden_dropout_prob)

        self.ranking_linear = torch.nn.Linear(config.hidden_size, 1)

        self.init_weights()

    

DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model_dir = "/bert_base-augmented-batch_size=128-lr=2e-5-max_gloss=6"

model = BertWSD.from_pretrained(model_dir)
tokenizer = BertTokenizer.from_pretrained(model_dir)
# add new special token
if '[TGT]' not in tokenizer.additional_special_tokens:
    tokenizer.add_special_tokens({'additional_special_tokens': ['[TGT]']})
    assert '[TGT]' in tokenizer.additional_special_tokens
    model.resize_token_embeddings(len(tokenizer))
    
model.to(DEVICE)
model.eval()
```