Dawn
Dawn
> Any update? All that's required is a minor adjustment to set `max_candidates=8` This should do the trick.
> Mind to submit a PR? Seems super minor.. so easy to mergeOn Dec 13, 2023, at 06:28, Andy ***@***.***> wrote: Any update? All that's required is a minor adjustment...
https://drive.google.com/file/d/1m8-EBPgi5MRubrm6iQjafK2QMHDBMSfJ/view
In the paper, they used the IMDB datasets, which are public. We can try downloading them and splitting them into 'forgot.csv' and 'retain.csv'
> @erikreed I got it working with the base (not chat) model. The evaluation scripts serve as good examples. E.g.: > > ```python > from transformers import AutoModelForCausalLM, AutoTokenizer >...