michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix cpu usage

Browse source
This commit is contained in:
C43H66N12O12S2 2022-09-19 18:18:33 +03:00 committed by AUTOMATIC1111
parent 422d17487f
commit cf28445f10
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/prompt_parser.py
View file

@ -114,7 +114,7 @@ def get_learned_conditioning(prompts, steps):
def reconstruct_cond_batch(c: ScheduledPromptBatch, current_step): def reconstruct_cond_batch(c: ScheduledPromptBatch, current_step):
res = torch.zeros(c.shape) res = torch.zeros(c.shape, device=shared.device, dtype=torch.half)
for i, cond_schedule in enumerate(c.schedules): for i, cond_schedule in enumerate(c.schedules):
target_index = 0 target_index = 0
for curret_index, (end_at, cond) in enumerate(cond_schedule): for curret_index, (end_at, cond) in enumerate(cond_schedule):
@ -123,7 +123,7 @@ def reconstruct_cond_batch(c: ScheduledPromptBatch, current_step):
break break
res[i] = cond_schedule[target_index].cond res[i] = cond_schedule[target_index].cond
return res.to(shared.device) return res