fix: surface trust_remote_code (#1139)

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

Author: akoumpa

nemo_automodel/components/models/biencoder/llama_bidirectional_model.py

@@ -528,6 +528,7 @@ def build(
         pooling: str = "avg",
         l2_normalize: bool = True,
         t: float = 1.0,
+        trust_remote_code: bool = False,
         **hf_kwargs,
     ):
         """
@@ -544,6 +545,7 @@ def build(
             pooling: Pooling strategy ('avg', 'cls', 'last', etc.)
             l2_normalize: Whether to L2 normalize embeddings
             t: Temperature for scaling similarity scores
+            trust_remote_code: Whether to trust remote code
             **hf_kwargs: Additional arguments passed to model loading
         """
 
@@ -575,7 +577,7 @@ def build(
         # Load model locally or from hub using selected model class
         if os.path.isdir(model_name_or_path):
             if share_encoder:
-                lm_q = ModelClass.from_pretrained(model_name_or_path, trust_remote_code=True, **hf_kwargs)
+                lm_q = ModelClass.from_pretrained(model_name_or_path, trust_remote_code=trust_remote_code, **hf_kwargs)
                 lm_p = lm_q
             else:
                 _qry_model_path = os.path.join(model_name_or_path, "query_model")
@@ -585,8 +587,8 @@ def build(
                     _qry_model_path = model_name_or_path
                     _psg_model_path = model_name_or_path
 
-                lm_q = ModelClass.from_pretrained(_qry_model_path, trust_remote_code=True, **hf_kwargs)
-                lm_p = ModelClass.from_pretrained(_psg_model_path, trust_remote_code=True, **hf_kwargs)
+                lm_q = ModelClass.from_pretrained(_qry_model_path, trust_remote_code=trust_remote_code, **hf_kwargs)
+                lm_p = ModelClass.from_pretrained(_psg_model_path, trust_remote_code=trust_remote_code, **hf_kwargs)
         else:
             # Load from hub
             lm_q = ModelClass.from_pretrained(model_name_or_path, **hf_kwargs)