最近刚开始研究Stable Diffusion model 想尝试一下fine-tuning任务。在本地加载模型后,爆显存了。。。于是不得已,在云平台租用服务器来进行训练。
创建服务器实例时,我直接选的的平台提供的算法镜像,这样可以免去配环境之苦:
然后,我在服务器运行程序的时候,在自己电脑上正常运行的程序它居然报错了:
OSError: CompVis/stable-diffusion-v1-4 does not appear to have a file named config.json
由于本人是第一次在linux系统下运行深度学习模型,以为是因为系统的问题,因为前面下载下来的模型文件很奇怪, 是一堆名为乱码的文件。
在网络上我也搜索不到问题的解决方案。经几天的排查,发现是服务器镜像环境中的transforms库的版本太低了,只需运行:
pip install --upgrade transformers
更新transformers库即可成功下载。