IKE to edit Llama2 on ZsRE and Reproducing Editing Performance
Hello,
- How can we edit llama2 on ZsRE with IKE method? I tried with
python run_zsre_llama2.py --editing_method=IKE --hparams_dir=../tutorial-notebooks/hparams/IKE/llama-7b --data_dir=./data
It shows the error : assert 'train_ds' in kwargs.keys() or print('IKE need train_ds(For getting In-Context prompt)')
Where and how should we change the code?
-
If we just want to reproduce the results of Editing Performance Table with the four metrics how can we do that?
-
CUDA out of memory error while running the code in EasyEdit_Example_IKE.ipynb
Is there any other way apart from using hugging face accelerator? Some way to reduce batch_size, etc?