awslabs / fortuna

A Library for Uncertainty Quantification.
https://aws-fortuna.readthedocs.io/en/latest/
Apache License 2.0
892 stars 46 forks source link

Fix initial position in hallucination pipeline #164

Closed gianlucadetommaso closed 11 months ago

gianlucadetommaso commented 11 months ago

Fix a bug: length of context (or text) is on dimension 1, not dimension 0. Furthermore, instead of init_pos, expose the number of final tokens n_final_tokens.