dsyxr369

训练到最后一步时，一直出现两个错误：
一、中间出的错误：
RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel:
size mismatch for
down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k.weight: copying a
param with shape torch.Size([320, 768]) from checkpoint, the shape in current
model is torch.Size([320, 1024]).
        ...
二、最后出的错误：
CalledProcessError: Command '['/usr/bin/python3', 'train_network.py',
以上两个错误都是在5.5. Start Training，最后一步时出现的错误
结束后，就是没有最终的训练模型文件
楼主有时间帮忙看看怎么能解决呢？万分感谢

[ 本帖最后由 dsyxr369 于 2023-4-13 19:02(GMT+8) 编辑 ]

dsyxr369

老大，依据你的指点，居然生成模型文件了，哈哈，谢谢

引用:

原帖由 内力煮火锅 于 2023-4-13 19:53 发表

2.1那里选择的是Stable-Diffusion-v1-5吗？
是的话，再看5.1那里，V2有没有打钩？注意这里V2不要打钩，因为在2.1那里选择的是v1-5。

dsyxr369

楼主老大，依据教程，走到第六步：用复制来的文件的路径替换掉下面的安装指令中的全部汉字，并将修改好的安装指令粘贴进Colab并运行时，
%pip install torch==1.13.1 cu117 torchvision==0.14.1 cu117 torchtext==0.14.1 torchaudio==0.13.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu117，提示：
ERROR: Could not find a version that satisfies the requirement cu117 (from versions: none)
ERROR: No matching distribution found for cu117，
最后提示错误：
RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install.
我的运行环境：windows10,python版本3.7.0,问题出在哪了呢？请都老大赐教

[ 本帖最后由 dsyxr369 于 2023-4-14 10:45(GMT+8) 编辑 ]

dsyxr369

引用:

原帖由 内力煮火锅 于 2023-4-14 16:10 发表

更新了安装指令，现在应该没问题了。
重新复制一遍现在的安装指令，再运行一遍试试。
记得先断开连接并删除运行时，然后再重新连接，然后从第二步在Colab的设置里打开显卡开始一步一步做下去。 ...

已能成功出图，关键词用不好，出图质量没有楼主的图质量丰富，清新自然。跪谢楼主老大