KeyError: 'labels' in generate_samples.py; doc_red about rebel HOT 4 CLOSED

Same issue as #29

Please check there.

Best,
Pere-Lluis.

from rebel.

Thanks for the quick reply.
This unfortunately doesn’t fix the issue for me.
In #29 the problem got resolved by setting pytorch-lightning verision to 1.1.7.
I was already using this version as specified in the requirements.txt.

As I ran into the same issue as #28 I changed the requirements.txt.
For me it only starts training with the following file:

omegaconf==2.0.6
pytorch-lightning==1.1.7
hydra-core==1.0.6
transformers
neptune-client==0.5.1
psutil==5.8.0
datasets==1.3.0
rouge-score==0.0.4
sacrebleu==1.5.0
wandb==0.10.26
streamlit==0.82.0

Which versions would you recommend @LittlePea13?

from rebel.

with the latest commit the issue with the labels should be fixed, I will update the reqs with a newer version of datasets but it needs a small change to the dataset files.

from rebel.

Thank you!

from rebel.