Emanuela Boros
commited on
Commit
·
0990522
1
Parent(s):
7b590c4
add confidence
Browse files- generic_nel.py +1 -1
generic_nel.py
CHANGED
|
@@ -95,7 +95,7 @@ class NelPipeline(Pipeline):
|
|
| 95 |
num_return_sequences=5,
|
| 96 |
max_new_tokens=30,
|
| 97 |
)
|
| 98 |
-
print(outputs)
|
| 99 |
# token_ids, scores = outputs.sequences, outputs.sequences_scores
|
| 100 |
# scores_tensor = scores.clone().detach()
|
| 101 |
# probabilities = torch.exp(scores_tensor)
|
|
|
|
| 95 |
num_return_sequences=5,
|
| 96 |
max_new_tokens=30,
|
| 97 |
)
|
| 98 |
+
# print(outputs)
|
| 99 |
# token_ids, scores = outputs.sequences, outputs.sequences_scores
|
| 100 |
# scores_tensor = scores.clone().detach()
|
| 101 |
# probabilities = torch.exp(scores_tensor)
|