ValueError: The repository for contains custom code which must be executed to ...的一种解决方案

Superstarimage 2024-09-12 15:31:01 阅读 89

博主在使用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, # 新增

)



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。