298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix routing_key crash

Browse Source
This commit is contained in:
drunkplato 2024-12-03 11:20:59 +00:00 committed by Ubuntu
parent 6ea9f911d3
commit 988246de7f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
memedeck.py
View File

@ -144,12 +144,12 @@ class MemedeckWorker:
prompt = payload["nodes"]
valid = self.validate_prompt(prompt)
routing_key = method.routing_key
workflow = 'training' if self.training_only else 'faceswap' if routing_key == 'faceswap-queue' else 'generation'
# Prepare task_info
prompt_id = str(uuid.uuid4())
outputs_to_execute = valid[2]
routing_key = method.routing_key
task_info = {
"workflow": workflow,
"prompt_id": prompt_id,
@ -431,8 +431,8 @@ class MemedeckWorker:
return
api_endpoint = '/generation/update' if task['workflow'] != 'training' else '/training/update'
self.logger.info(f"[memedeck]: sending to api: {api_endpoint}")
self.logger.info(f"[memedeck]: data: {data}")
# self.logger.info(f"[memedeck]: sending to api: {api_endpoint}")
# self.logger.info(f"[memedeck]: data: {data}")
try:
# this request is not sending properly for faceswap
post_func = partial(requests.post, f"{self.api_url}{api_endpoint}", json=data)