Update model.py
Browse files
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,
|