今天在 Stable Diffusion 使用时新添加了一个.safetensors文件的 VAE 模型,但是选择后不能加载,自动返回以前的 VAE 模型,再次运行生成图片时出现了以下报错:
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)。
无论怎么做都不能生成图片了,无奈之下只有把这个 VAE 模型删除,并重新启动 Stable Diffusion 才能正常使用。
网上说的在启动文件 bat 中放置“–disable-safe-unpickle”命令也不行。
不知道是不是模型问题,而且这个 VAE 模型很大,也许不该放在 VAE 下,应该放在大模型文件下。
发表回复