Spaces:
Sleeping
Sleeping
Update run_eval.py
Browse files- run_eval.py +10 -2
run_eval.py
CHANGED
@@ -49,10 +49,18 @@ for cfg in CONFIGS:
|
|
49 |
|
50 |
# Try causal first, fallback to encoder
|
51 |
try:
|
52 |
-
base_model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
|
|
|
|
|
53 |
is_encoder = False
|
54 |
except:
|
55 |
-
base_model = AutoModelForSequenceClassification.from_pretrained(
|
|
|
|
|
|
|
|
|
56 |
is_encoder = True
|
57 |
|
58 |
peft_model = PeftModel.from_pretrained(base_model, adapter_repo)
|
|
|
49 |
|
50 |
# Try causal first, fallback to encoder
|
51 |
try:
|
52 |
+
base_model = AutoModelForCausalLM.from_pretrained(
|
53 |
+
base_model_id,
|
54 |
+
trust_remote_code=True,
|
55 |
+
use_safetensors=True
|
56 |
+
)
|
57 |
is_encoder = False
|
58 |
except:
|
59 |
+
base_model = AutoModelForSequenceClassification.from_pretrained(
|
60 |
+
base_model_id,
|
61 |
+
trust_remote_code=True,
|
62 |
+
use_safetensors=True
|
63 |
+
)
|
64 |
is_encoder = True
|
65 |
|
66 |
peft_model = PeftModel.from_pretrained(base_model, adapter_repo)
|