Open sleeper opened 4 months ago
pip install evaluate?
OK, I had to add: evaluate, torch, transformers, nltk, rouge-score and transformers[torch]. I'm still facing some errors (but seems mainly link to the architecture selection, as it complains on the fact I do not have a CUDA, XPU, NPU .. device).
Shouldn't these dependencies be part of the pyproject.toml
file?
We don't want to depend on them since >90% of users don't need BootstrapFinetune yet. But we do need to raise a nice error that says please install them.
Indeed. And we can also add it to the documentation :)
The file
dsp/modules/finetuning/finetune_hf.py
has a dependency on the module evaluate coming from the copy of https://www.philschmid.de/fine-tune-flan-t5#3-fine-tune-and-evaluate-flan-t5, but this dependency is never installed, leading to an error when playing with fine-tuning.