stable diffusion移植1684

Viewed 429

您好,我按照这个帖子操作:https://zhuanlan.zhihu.com/p/579395297
发生如下报错,请帮忙看看是什么问题。

1、下载SD1-4官方的onnx textencoder模型,并用如下命令转换
python3 -m bmneto --model=./text_encoder.onnx
--outdir="./"
--target="BM1684"
--shapes="1,77"
--opt=1
--cmp=false
--net_name="text_encoder"
--descs="[0,int64,0,49409]"
报错:image.png

2、修改命令为:
python3 -m bmneto --model=./text_encoder.onnx
--outdir="./"
--target="BM1684"
--shapes="1,77"
--opt=1
--cmp=false
--net_name="text_encoder"
--descs="[0,int32,0,49409]"
报错:
image.png

3、在迁移VAE模型时,运行如下脚本:
image.png
报错:
image.png

1 Answers

你好,可以提供一下trace的模型吗?以百度网盘形式或者其他方式都可以。

没有trace的模型呀。。因为运行报错了。。上面描述中的第1、2是直接基于官方提供的onnx模型,第3是基于官方pytorch模型,在trace的时候报错了。。

  1. 好的,onnx模型,我待会再去验证一下。
  2. trace的报错是因为nograd不一定能将所有梯度消除,这是因为对于modulelist和list中的model不一定生效。可以通过如下方式解决:
    for param in pipe.vae.parameters():
    param.requires_grad = False