You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

使用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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.10 22:32:07