Mdrnfox commited on
Commit
adef9e5
·
verified ·
1 Parent(s): ddcd17e

Update run_eval.py

Browse files
Files changed (1) hide show
  1. 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(base_model_id)
 
 
 
 
53
  is_encoder = False
54
  except:
55
- base_model = AutoModelForSequenceClassification.from_pretrained(base_model_id)
 
 
 
 
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)