Random global ID

This commit is contained in:
City
2023-10-07 20:27:23 +02:00
parent 79a5622d32
commit 85a9cf5f1b
+12 -2
View File
@@ -8,6 +8,16 @@ from PIL import Image
from copy import deepcopy from copy import deepcopy
# set global ID once for entire session
try: GID
except NameError:
GID = ''.join(random.choice("abcdefghijklmnopqrstupvxyz") for x in range(5))
print(f"NetDist: Set session ID to '{GID}'")
def get_client_id():
global GID
return(f"netdist-{GID}")
class FetchRemote(): class FetchRemote():
def __init__(self): def __init__(self):
pass pass
@@ -104,7 +114,7 @@ class QueueRemoteChainStart:
"prompt": prompt, "prompt": prompt,
"current_seed": seed+batch, "current_seed": seed+batch,
"current_batch": batch, "current_batch": batch,
"job_id": f"netdist-{time.time()}" "job_id": f"{get_client_id()}-{int(time.time()*1000*1000)}"
} }
return(remote_chain,) return(remote_chain,)
@@ -246,7 +256,7 @@ class QueueRemote:
data = { data = {
"prompt": prompt, "prompt": prompt,
"client_id": "netdist", "client_id": get_client_id(),
"extra_data": { "extra_data": {
"job_id": remote_info["job_id"], "job_id": remote_info["job_id"],
} }