Hello,

1. 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?

2. If we just want to reproduce the results of Editing Performance Table with the four metrics how can we do that?

3. 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?