IrisRainbowNeko / DreamArtist-stable-diffusion

stable diffusion webui with contrastive prompt tuning
876 stars 53 forks source link

Run on GPU? #22

Open TumnusB opened 1 year ago

TumnusB commented 1 year ago

My rig has a 4090 and I'm getting a error about CPU? same one as https://github.com/7eu7d7/DreamArtist-stable-diffusion/issues/16 so not sure if duplicate or not?

Loaded a total of 2 words embeddings. Embeddings: Elins-neg, Elins Training at rate of 0.005 until step 100000 Preparing dataset... 100%|██████████████████████████████████████████████████████████████████████████████████| 12/12 [00:12<00:00, 1.07s/it] 0%| | 0/100000 [00:03<?, ?it/s] Applying xformers cross attention optimization. Error completing request Arguments: ('Elins', '0.005', 1, 'C:\Users\PERSON\Pictures\ELINS', 'dream_artist', 768, 768, 100000, 250, 250, 'E:\DreamArtist-stable-diffusion\textual_inversion_templates\style_filewords.txt', True, False, '', '', 20, 0, 7, -1.0, 512, 512, 5.0, '', True, True, 1, 1, 1.0, 25.0, 1.0, 25.0, 0.9, 0.999) {} Traceback (most recent call last): File "E:\DreamArtist-stable-diffusion\modules\ui.py", line 185, in f res = list(func(*args, kwargs)) File "E:\DreamArtist-stable-diffusion\webui.py", line 54, in f res = func(*args, kwargs) File "E:\DreamArtist-stable-diffusion\modules\dream_artist\ui.py", line 36, in train_embedding embedding, filename = modules.dream_artist.cptuning.train_embedding(args) File "E:\DreamArtist-stable-diffusion\modules\dream_artist\cptuning.py", line 436, in train_embedding output = shared.sd_model(x, c_in, scale=cfg_scale) File "E:\DreamArtist-stable-diffusion\venv\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(input, kwargs) File "E:\DreamArtist-stable-diffusion\repositories\stable-diffusion\ldm\models\diffusion\ddpm.py", line 879, in forward return self.p_losses(x, c, t, *args, kwargs) File "E:\DreamArtist-stable-diffusion\modules\dream_artist\cptuning.py", line 286, in p_losses_hook 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)