lucweber commited on
Commit
ce118fe
·
verified ·
1 Parent(s): 7f82063

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -0
model.py CHANGED
@@ -90,6 +90,9 @@ class CausalLMForRegression(nn.Module):
90
  cfg = AutoConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
91
  cfg.output_hidden_states = True
92
 
 
 
 
93
  backbone = Qwen3ForCausalLM.from_pretrained(
94
  pretrained_model_name_or_path,
95
  *model_args,
 
90
  cfg = AutoConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
91
  cfg.output_hidden_states = True
92
 
93
+ if "trust_remote_code" in kwargs:
94
+ _ = kwargs.pop("trust_remote_code")
95
+
96
  backbone = Qwen3ForCausalLM.from_pretrained(
97
  pretrained_model_name_or_path,
98
  *model_args,