ModuleNotFoundError: No module named 'peft' Sagemaker BLOOM Finetuning

Amazon SageMaker
1

Hi,
I am trying to replicate this tutorial: Train and Deploy BLOOM with Amazon SageMaker and PEFT but I am getting this error in training job:

ErrorMessage “ModuleNotFoundError: No module named ‘peft’”

2

It looks like that your source_dir is missing the requirements.txt which installs peft in the beginning of the training.
Make sure to also include the infernece.py script.

2 Likes
3

Thanks for the reply. Yes, this was the issue. The requirements.txt was outside the directory.