typo fix for get_probabilities (#4)

- typo fix for get_probabilities (19f5a83879ad41eb66c3f613202a9ff743d49b85)


Co-authored-by: Manish Nagireddy <[email protected]>

README.md

@@ -77,7 +77,7 @@ def parse_output(output, input_len):
         list_index_logprobs_i = [torch.topk(token_i, k=nlogprobs, largest=True, sorted=True)
                                  for token_i in list(output.scores)[:-1]]
         if list_index_logprobs_i is not None:
-            prob = get_probablities(list_index_logprobs_i)
+            prob = get_probabilities(list_index_logprobs_i)
             prob_of_risk = prob[1]
 
     res = tokenizer.decode(output.sequences[:,input_len:][0],skip_special_tokens=True).strip()
@@ -90,7 +90,7 @@ def parse_output(output, input_len):
 
     return label, prob_of_risk.item()
 
-def get_probablities(logprobs):
+def get_probabilities(logprobs):
     safe_token_prob = 1e-50
     unsafe_token_prob = 1e-50
     for gen_token_i in logprobs: