sd-webui/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/sd-webui/stable-diffusion-webui.git synced 2024-12-14 23:02:00 +03:00
Code Issues Projects Releases Wiki Activity

JobManager: Print true exception callstack (#590)

Browse Source 
* JobManager: Print true exception callstack

* Allow empty prompts

Bug #564

Co-authored-by: hlky <106811348+hlky@users.noreply.github.com>
This commit is contained in:
cobryan05 2022-09-03 18:21:46 -05:00 committed by GitHub
parent 7b352ad038
commit d8a323ebc6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frontend/job_manager.py
View File

@ -8,6 +8,7 @@ from dataclasses import dataclass, field
from functools import partial
from PIL.Image import Image
import uuid
import traceback
@dataclass(eq=True, frozen=True)
@ -231,7 +232,7 @@ class JobManager:
outputs = job_info.func(*job_info.inputs, job_info=job_info)
except Exception as e:
job_info.job_status = f"Error: {e}"
print(f"Exception processing job {job_info}: {e}")
print(f"Exception processing job {job_info}: {e}\n{traceback.format_exc()}")
outputs = []
# Filter the function output for any removed outputs

4
webui.py
View File

@ -782,7 +782,7 @@ def process_images(
uses_random_seed_loopback=False, sort_samples=True, write_info_files=True, write_sample_info_to_log_file=False, jpg_sample=False,
variant_amount=0.0, variant_seed=None,imgProcessorTask=False, job_info: JobInfo = None):
"""this is the main loop that both txt2img and img2img use; it calls func_init once inside all the scopes and func_sample once per batch"""
assert prompt is not None
prompt = prompt or ''
torch_gc()
# start time after garbage collection (or before?)
start_time = time.time()
@ -1508,7 +1508,7 @@ def split_weighted_subprompts(input_string, normalize=True):
weight_sum = sum(map(lambda x: x[1], parsed_prompts))
if weight_sum == 0:
print("Warning: Subprompt weights add up to zero. Discarding and using even weights instead.")
equal_weight = 1 / len(parsed_prompts)
equal_weight = 1 / (len(parsed_prompts) or 1)
return [(x[0], equal_weight) for x in parsed_prompts]
return [(x[0], x[1] / weight_sum) for x in parsed_prompts]