Spaces:
Sleeping
Sleeping
config_overrides parameter to remove problematic bias settings
Browse files- 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(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|