jdchang
/

test_rm_8b

Feature Extraction

Model card Files Files and versions Community

jdchang commited on Nov 18

Commit

37325f3

•

1 Parent(s): a25fa33

Update hf_utils.py

Files changed (1) hide show

hf_utils.py +1 -1

hf_utils.py CHANGED Viewed

@@ -249,7 +249,7 @@ class AutoModelForCausalLMWithRM(PreTrainedModel):
         use_safetensors: Optional[bool] = None,
         **kwargs: Any,
     ) -> PreTrainedModel:
-        trust_remote_code = kwargs.pop('trust_remote_code', None)
         use_flash_attention_2 = kwargs.pop('use_flash_attention_2', False)
         return_lm_logits = kwargs.pop('return_lm_logits', False)
         load_in_8bit = kwargs.pop('load_in_8bit', False)

         use_safetensors: Optional[bool] = None,
         **kwargs: Any,
     ) -> PreTrainedModel:
+        trust_remote_code = kwargs.pop('trust_remote_code', True)
         use_flash_attention_2 = kwargs.pop('use_flash_attention_2', False)
         return_lm_logits = kwargs.pop('return_lm_logits', False)
         load_in_8bit = kwargs.pop('load_in_8bit', False)