解决sd-webui中rembg插件报错及onnxruntime重装指南

针对sd-webui中rembg插件使用报错的问题,以下是根据参考资料提供的详细解决方案:

图片[1]_解决sd-webui中rembg插件报错及onnxruntime重装指南_知途无界

问题背景

stable-diffusion-webui-rembg是一个非常实用的sd-webui插件,它能够实现一键抠图,对于主体明确、线条简单的图片效果相当好。然而,在安装并使用该插件时,可能会遇到一些报错问题。

报错问题及解决方案

1. 模型下载问题

由于国内访问GitHub可能存在困难,建议预先下载好模型。具体下载地址可以在AKI一键启动包的控制台中找到。如果已经下载了一个通用模型(如isnet-general-use),可以将其放入指定的目录(如D:\sd-webui-aki-v4.7.cache\u2net)。注意,这里的路径需要替换成你实际的安装路径。

2. 运行时报错

在实际运行rembg插件时,可能会遇到如下报错:

RuntimeError: D:\a\_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1106 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "D:\sd-webui-aki-v4.7\python\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"

这个报错通常是由于onnxruntime库的安装或配置问题导致的。以下是一个有效的解决方案:

  • 首先,打开命令行工具,并切换到sd-webui的python环境目录下(如D:\sd-webui-aki-v4.7\python)。
  • 可选步骤:升级pip到最新版本。使用以下命令:
python.exe -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 这里使用了清华大学的pypi镜像源,以提高下载速度。
  • 核心处置步骤:卸载当前的onnxruntime-gpu版本(如果安装了的话),然后重新安装onnxruntime的CPU版本。使用以下命令:
python.exe .\Scripts\pip3.exe uninstall onnxruntime-gpu  
python.exe .\Scripts\pip3.exe install onnxruntime --user -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 注意,这里同样使用了清华大学的pypi镜像源。如果你确定需要GPU支持,并且你的系统已经正确配置了CUDA和cuDNN,你也可以尝试安装onnxruntime-gpu版本,但需要先确保你的系统环境满足GPU加速的条件。

测试效果

完成上述步骤后,你可以尝试重新运行sd-webui并使用rembg插件进行抠图操作,以检查问题是否已解决。

注意事项

  • 在进行上述操作时,请确保你的sd-webui和rembg插件都是最新版本,以避免因版本不兼容导致的问题。
  • 如果你在使用GPU版本的onnxruntime时遇到问题,可以尝试切换到CPU版本进行尝试。
  • 如果你在下载或安装模型时遇到问题,可以尝试检查网络连接或尝试使用其他下载源。

希望以上解决方案能够帮助你解决sd-webui中rembg插件使用报错的问题。如果还有其他问题或需要进一步的帮助,请随时联系。

© 版权声明
THE END
喜欢就点个赞,支持一下吧!
点赞12 分享
评论 抢沙发
头像
欢迎您留下评论!
提交
头像

昵称

取消
昵称表情代码图片

    暂无评论内容