stable-cascade 文生图模型diffusers使用案例

作者：weixin_40725706 | 2024-02-20 19:46:45

踩

参考：https://huggingface.co/stabilityai/stable-cascade

下载：需要分别下载两个模型stabilityai/stable-cascade-prior与stabilityai/stable-cascade

diffusers版本也要指定安装：

pip install git+https://github.com/kashif/diffusers.git@wuerstchen-v3
1

代码

1）gpu版本t4 15g卡运行显存不够，大于15-6g的显卡可以；
2）报错需要修改：
1、embedding.1.weight expected shape tensor(…, device=‘meta’, size=(320, 64, 1, 1)), but got torch.Size([320, 16, 1, 1]).
参考修改： open “stable-cascade\decoder\config.js

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/120740