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

博主在使用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, # 新增
)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值