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, # 新增
)
上一篇: Java 和 .NET Core 在企业级应用开发中各有什么优势和劣势?看看AI是怎么回答的
下一篇: 【机器学习】pytorch 常用函数解析
本文标签
ValueError: The repository for contains custom code which must be executed to ...的一种解决方案
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。