1580158
/
stable-diffusion-webui-distributed
mirror of https://github.com/papuSpartan/stable-diffusion-webui-distributed.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix for controlnet

master^2
papuSpartan 2024-10-05 03:32:42 -05:00
parent 979d559803
commit 23ec02c229
No known key found for this signature in database
GPG Key ID: CA376082283AF69A
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/distributed.py
View File

@ -128,7 +128,7 @@ class DistributedScript(scripts.Script):
gallery_index = num_local + num_injected + i # zero-indexed point of image in total gallery
job.gallery_map.append(gallery_index) # so we know where to edit infotext
pp.images.append(image)
logger.debug(f"image {gallery_index + 1}/{self.world.num_total()}")
logger.debug(f"image {gallery_index + 1}/{self.world.num_gallery()}")
def update_gallery(self, pp, p):
"""adds all remotely generated images to the image gallery after waiting for all workers to finish"""

4
scripts/spartan/world.py
View File

@ -286,8 +286,8 @@ class World:
return num_images
def num_total(self) -> int:
"""how many images should be returned in total, so including local generations and its grid"""
def num_gallery(self) -> int:
"""How many images should appear in the gallery. This includes local generations and a grid(if enabled)"""
return self.num_requested() * self.p.n_iter + shared.opts.return_grid