现在跑图是没问题了,但是跑训练的时候又报错了,大佬能帮忙看看吗
Preparing dataset...
100% 134/134 [00:18<00:00, 7.27it/s]
0% 0/5000 [00:00<?, ?it/s]Traceback (most recent call last):
File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/textual_inversion/textual_inversion. py", line 328, in train_embedding
loss = shared.sd_model(x, c)[0] / gradient_step
File "/usr/local/lib/python3.8/dist-package
网页链接 nn/modules/module. py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/models/diffusion/ddpm. py", line 846, in forward
return self.p_losses(x, c, t, *args, **kwargs)
File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/models/diffusion/ddpm. py", line 903, in p_losses
logvar_t = self.logvar[t].to(self.device)
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)