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__index_select)
When try to use 512-base-ema.ckpt
Traceback (most recent call last):
File "C:\Users\ZeroCool22\Desktop\Auto\modules\call_queue.py", line 45, in f
res = list(func(*args, **kwargs))
File "C:\Users\ZeroCool22\Desktop\Auto\modules\call_queue.py", line 28, in f
res = func(*args, **kwargs)
File "C:\Users\ZeroCool22\Desktop\Auto\modules\txt2img.py", line 49, in txt2img
processed = process_images(p)
File "C:\Users\ZeroCool22\Desktop\Auto\modules\processing.py", line 430, in process_images
res = process_images_inner(p)
File "C:\Users\ZeroCool22\Desktop\Auto\modules\processing.py", line 520, in process_images_inner
uc = prompt_parser.get_learned_conditioning(shared.sd_model, negative_prompts, p.steps)
File "C:\Users\ZeroCool22\Desktop\Auto\modules\prompt_parser.py", line 138, in get_learned_conditioning
conds = model.get_learned_conditioning(texts)
File "C:\Users\ZeroCool22\Desktop\Auto\repositories\stable-diffusion-stability-ai\ldm\models\diffusion\ddpm.py", line 665, in get_learned_conditioning
c = self.cond_stage_model.encode(c)
File "C:\Users\ZeroCool22\Desktop\Auto\repositories\stable-diffusion-stability-ai\ldm\modules\encoders\modules.py", line 131, in encode
return self(text)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\ZeroCool22\Desktop\Auto\repositories\stable-diffusion-stability-ai\ldm\modules\encoders\modules.py", line 121, in forward
outputs = self.transformer(input_ids=tokens, output_hidden_states=self.layer=="hidden")
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\transformers\models\clip\modeling_clip.py", line 722, in forward
return self.text_model(
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\transformers\models\clip\modeling_clip.py", line 632, in forward
hidden_states = self.embeddings(input_ids=input_ids, position_ids=position_ids)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\transformers\models\clip\modeling_clip.py", line 165, in forward
inputs_embeds = self.token_embedding(input_ids)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\modules\sparse.py", line 158, in forward
return F.embedding(
File "C:\Users\ZeroCool22\Desktop\Auto\venv\lib\site-packages\torch\nn\functional.py", line 2199, in embedding
return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
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__index_select)