naver
/

splade-code-06B

Feature Extraction

sentence-transformers

text-generation

text-embeddings-inference

Model card Files Files and versions

maxoul commited on Feb 24

Commit

450c551

·

verified ·

1 Parent(s): 1095893

Update utils.py

Files changed (1) hide show

utils.py +2 -1

utils.py CHANGED Viewed

@@ -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
 ):
     """
     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,
     )

 def get_decoder_model(
+    model_name_or_path: str, attn_implementation: str, bidirectional: bool, base_cfg, token=None
 ):
     """
     base_cfg is the pretrained config of the underlying model
         config=base_cfg,
         torch_dtype=torch.bfloat16,
         attn_implementation=attn_implementation,
+        token=token,
     )