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

protect against cyclical configurations

master
unknown 2024-05-11 21:57:02 -05:00
parent 424a1c8858
commit 56d667dfe8
No known key found for this signature in database
GPG Key ID: CA376082283AF69A
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/distributed.py
View File

@ -62,7 +62,9 @@ class DistributedScript(scripts.Script):
return scripts.AlwaysVisible return scripts.AlwaysVisible
def ui(self, is_img2img): def ui(self, is_img2img):
self.world.load_config() if not is_img2img: # prevents loading twice for no reason
self.world.load_config()
extension_ui = UI(world=self.world) extension_ui = UI(world=self.world)
# root, api_exposed = extension_ui.create_ui() # root, api_exposed = extension_ui.create_ui()
components = extension_ui.create_ui() components = extension_ui.create_ui()

8
scripts/spartan/world.py
View File

@ -145,6 +145,14 @@ class World:
Worker: The worker object. Worker: The worker object.
""" """
# protect against user trying to make cyclical setups and connections
is_master = kwargs.get('master')
if is_master is None or not is_master:
m = self.master()
if kwargs['address'] == m.address and kwargs['port'] == m.port:
logger.error(f"refusing to add worker {kwargs['label']} as its socket definition({m.address}:{m.port}) matches master")
return None
original = self[kwargs['label']] # if worker doesn't already exist then just make a new one original = self[kwargs['label']] # if worker doesn't already exist then just make a new one
if original is None: if original is None:
new = Worker(**kwargs) new = Worker(**kwargs)