Update utils.py

utils.py

@@ -116,7 +116,7 @@ def prepare_tokenizer(tokenizer_name: str, padding_side="right"):
 
 
 def get_decoder_model(
-    model_name_or_path: str, attn_implementation: str, bidirectional: bool, base_cfg
+    model_name_or_path: str, attn_implementation: str, bidirectional: bool, base_cfg, token=None
 ):
     """
     base_cfg is the pretrained config of the underlying model
@@ -135,6 +135,7 @@ def get_decoder_model(
         config=base_cfg,
         torch_dtype=torch.bfloat16,
         attn_implementation=attn_implementation,
+        token=token,
     )