使用AutoTrain DreamBooth Colab遇错误:无法从diffusers.utils导入randn_tensor
解决AutoTrain DreamBooth Colab中
randn_tensor导入错误的方法 这个错误是因为diffusers库版本和AutoTrain脚本不兼容导致的——新版本diffusers调整了randn_tensor的存放位置,而笔记本里的代码还是按旧路径导入。可以按以下步骤解决:
方法一:锁定兼容的diffusers版本(推荐)
在Colab的代码单元格中,找到安装AutoTrain或diffusers的步骤,修改为指定版本安装:
!pip install diffusers==0.20.2
如果原笔记本里是通过AutoTrain依赖安装diffusers,也可以直接在AutoTrain安装命令后追加版本指定:
!pip install autotrain-advanced diffusers==0.20.2
修改后,先点击顶部菜单栏「运行时」→「重启运行时」,再重新执行所有代码单元格。
方法二:修改导入路径(针对新版本diffusers)
如果不想降级diffusers,找到报错的代码文件(通常是AutoTrain内部调用diffusers的脚本),把导入语句:
from diffusers.utils import randn_tensor
替换为:
from diffusers.utils.torch_utils import randn_tensor
不过这种方法需要定位到具体文件修改,在Colab里操作相对麻烦,更推荐用方法一。
内容的提问来源于stack exchange,提问作者SKB
相关产品推荐
相关产品推荐

