search

guaneec / custom-diffusion-webui

An unofficial implementation of Custom Diffusion for Automatic1111's WebUI.
GNU Affero General Public License v3.0
70 stars 6 forks source link

Delta error #8

Open hypernewb opened 1 year ago

hypernewb commented 1 year ago

Training textual inversion [Epoch 99: 1/1] loss: 0.1027512: 1%| | 99/8000 [02:Traceback (most recent call last): File "/storage/stable-diffusion/stable-diffusion-webui/extensions/custom-diffusion-webui/cd_modules/custom_diffusion.py", line 231, in _train_embedding save_deltas(last_saved_delta) File "/storage/stable-diffusion/stable-diffusion-webui/extensions/custom-diffusion-webui/cd_modules/custom_diffusion.py", line 154, in save_deltas delta = Delta(tensors={k: kvs[k] - kvs_bak[k] for k in kvs}) File "/storage/stable-diffusion/stable-diffusion-webui/extensions/custom-diffusion-webui/cd_modules/custom_diffusion.py", line 154, in delta = Delta(tensors={k: kvs[k] - kvs_bak[k] for k in kvs}) RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

Any help with this issue? when trying to save in log dir it fails