gpt2-xl in run_santinization_llama2.py

[kailinjiang]

python3 examples/run_santinization_llama2.py --editing_method FT --hparams_dir /home/jiangkailin/project/ice_editing/hparams/FT/gpt2-xl.yaml --data_dir /home/jiangkailin/project/ice_editing/knowedit_data/Sanitation --specify_answer "julius caesar"

Traceback (most recent call last): File "/home/jiangkailin/project/EasyEdit-main/examples/run_santinization_llama2.py", line 614, in exp = Experimenter(args=args, template="Question:{}\nAnswer:") File "/home/jiangkailin/project/EasyEdit-main/examples/run_santinization_llama2.py", line 461, in init self.cache_train_before_editing(replace=True, max_new_tokens=20) File "/home/jiangkailin/project/EasyEdit-main/examples/run_santinization_llama2.py", line 75, in cache_train_before_editing {q:self.filter_prefix(q, a) for q, a in zip(dataset["prompt"], ans)} File "/home/jiangkailin/project/EasyEdit-main/examples/run_santinization_llama2.py", line 75, in {q:self.filter_prefix(q, a) for q, a in zip(dataset["prompt"], ans)} File "/home/jiangkailin/project/EasyEdit-main/examples/run_santinization_llama2.py", line 53, in filter_prefix assert prefix in generate AssertionError

[zxlzr]

Sorry the script your are running (run_santinization_llama2.py) is for llama2, it do not support gpy2-xl.