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
# 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():
def __init__(self):
pass
@@ -104,7 +114,7 @@ class QueueRemoteChainStart:
"prompt": prompt,
"current_seed": seed+batch,
"current_batch": batch,
"job_id": f"netdist-{time.time()}"
"job_id": f"{get_client_id()}-{int(time.time()*1000*1000)}"
}
return(remote_chain,)
@@ -246,7 +256,7 @@ class QueueRemote:
data = {
"prompt": prompt,
"client_id": "netdist",
"client_id": get_client_id(),
"extra_data": {
"job_id": remote_info["job_id"],
}