readme extras for VRAM for

added missing packages to requirements for #74
add support for negative numbers in X/Y plot (plus ranges) #73
changed progressbar to work properly with custom modes
This commit is contained in:
AUTOMATIC
2022-09-06 10:11:25 +03:00
parent beece7d85c
commit 395f170597
7 changed files with 45 additions and 11 deletions

View File

@@ -153,7 +153,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
with torch.no_grad(), precision_scope("cuda"), ema_scope():
p.init()
state.job_count = p.n_iter
if state.job_count == -1:
state.job_count = p.n_iter
for n in range(p.n_iter):
if state.interrupted:

View File

@@ -54,6 +54,7 @@ class State:
self.job_no += 1
self.sampling_step = 0
state = State()
artist_db = modules.artists.ArtistsDatabase(os.path.join(script_path, 'artists.csv'))

View File

@@ -140,7 +140,10 @@ def check_progress_call():
if shared.state.job_count == 0:
return ""
progress = shared.state.job_no / shared.state.job_count
progress = 0
if shared.state.job_count > 0:
progress += shared.state.job_no / shared.state.job_count
if shared.state.sampling_steps > 0:
progress += 1 / shared.state.job_count * shared.state.sampling_step / shared.state.sampling_steps