ValueError: --caption_column‘ value ‘text‘ needs to be one of: image

使用diffusers在自己的数据集上进行训练报错,刚开始以为是自己的数据集格式不对,尝试了hugging face(datasets-image_load)中的各种数据格式都不行,网上搜索一圈,也没找到好的解决办法,最后死马当做活马医,尝试降低datasets版本(2.14.5=> 2.13.0),竟然凑效了。。。版本不兼容是真的难顶:(,很不容易debug,即便debug也没啥教育意义。。。希望给大家避坑

你可能感兴趣的:(图像生成,stable,diffusion,深度学习)