ModuleNotFoundError: No module named 'datasets' when running generation

[lilycai2000]

Hi, when I was running 'python generate.py --ckpt_path ${model_path} --in_image examples/sample0.png --in_mask examples/mask0.png --text wizards --output_dir ${output_dir}', there was an error: 'ModuleNotFoundError: No module named 'datasets''. I didn't see any dataset folder in this repo, any idea on how to solve this problem?

Thanks!

[Question406]

Hi, I think this module should be the datasets library from huggingface, https://huggingface.co/docs/datasets/index. I checked the requirements and find that it is included. Did you check if the environment is correctly installed?

[lilycai2000]

thanks it's solved!!