Update scorer_pred_local.py
Browse files- scorer_pred_local.py +0 -1
scorer_pred_local.py
CHANGED
|
@@ -20,7 +20,6 @@ if __name__ == "__main__":
|
|
| 20 |
parser.add_argument('--do-score-filter', action='store_true', default=False, help='do score filter or not', dest='do_score_filter')
|
| 21 |
args = parser.parse_args()
|
| 22 |
|
| 23 |
-
# model_dir = '/share/project/ldwang/Aquila3/quality_scorer_base_from_qwen15_0_5b_labeled_by_deepspeek-v2'
|
| 24 |
model_dir = args.scorer_model_path
|
| 25 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 26 |
model_dir,
|
|
|
|
| 20 |
parser.add_argument('--do-score-filter', action='store_true', default=False, help='do score filter or not', dest='do_score_filter')
|
| 21 |
args = parser.parse_args()
|
| 22 |
|
|
|
|
| 23 |
model_dir = args.scorer_model_path
|
| 24 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 25 |
model_dir,
|