博主在使用diffusers导入自定义的数据集训练stable diffusion时,发现自定义数据集中的加载脚本不起作用(加载自定义数据集的方法见博文),并且报如下错误:
ValueError: The repository for contains custom code which must be
executed to correctly load the model. You can inspect the repository
content at https://hf.co/xxx。
经过多种尝试后,最终得到一种解决方法,即:在load_dataset()方法的形参中额外增加一条参数trust_remote_code=True,
dataset = load_dataset(
args.dataset_name,
args.dataset_config_name,
cache_dir=args.cache_dir,
data_dir=args.train_data_dir,
trust_remote_code=True, # 新增
)