Tonic commited on
Commit
62bb2f0
·
verified ·
1 Parent(s): c339ecd

config_overrides parameter to remove problematic bias settings

Browse files
Files changed (1) hide show
  1. tasks/text.py +15 -2
tasks/text.py CHANGED
@@ -64,8 +64,21 @@ async def evaluate_text(request: TextEvaluationRequest):
64
  # Initialize tokenizer
65
  tokenizer = AutoTokenizer.from_pretrained(path_tokenizer)
66
 
67
- # Initialize model
68
- model = AutoModelForSequenceClassification.from_pretrained(path_model).half().to(device)
 
 
 
 
 
 
 
 
 
 
 
 
 
69
 
70
  # Set model to evaluation mode
71
  model.eval()
 
64
  # Initialize tokenizer
65
  tokenizer = AutoTokenizer.from_pretrained(path_tokenizer)
66
 
67
+ # Initialize model with specific configuration
68
+ model = AutoModelForSequenceClassification.from_pretrained(
69
+ path_model,
70
+ trust_remote_code=True,
71
+ num_labels=8,
72
+ problem_type="single_label_classification",
73
+ ignore_mismatched_sizes=True,
74
+ torch_dtype=torch.float16,
75
+ config_overrides={
76
+ "norm_bias": None, # Remove bias parameter
77
+ "classifier_bias": None,
78
+ "attention_bias": None,
79
+ "mlp_bias": None
80
+ }
81
+ ).to(device)
82
 
83
  # Set model to evaluation mode
84
  model.eval()