Update factual/RaTEScore/scorer.py
Browse files
factual/RaTEScore/scorer.py
CHANGED
@@ -18,7 +18,7 @@ class RaTEScore:
|
|
18 |
bert_model="Angelakeke/RaTE-NER-Deberta",
|
19 |
eval_model='FremyCompany/BioLORD-2023-C',
|
20 |
batch_size=1,
|
21 |
-
use_gpu=
|
22 |
visualization_path=None,
|
23 |
affinity_matrix="long",
|
24 |
):
|
|
|
18 |
bert_model="Angelakeke/RaTE-NER-Deberta",
|
19 |
eval_model='FremyCompany/BioLORD-2023-C',
|
20 |
batch_size=1,
|
21 |
+
use_gpu=False,
|
22 |
visualization_path=None,
|
23 |
affinity_matrix="long",
|
24 |
):
|