@andre15silva,
After executing bash run_llama_pred.sh, iam getting Exception has occurred: KeyError 'buggy_code'.
I have used the same dataset "ir4xor2" for traning.
Kindly let me know which dataset needs to be used for successfully running bash run_llama_pred.sh