xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-09 20:07:08 +08: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"] prompt = payload["nodes"]
valid = self.validate_prompt(prompt) 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' workflow = 'training' if self.training_only else 'faceswap' if routing_key == 'faceswap-queue' else 'generation'
# Prepare task_info # Prepare task_info
prompt_id = str(uuid.uuid4()) prompt_id = str(uuid.uuid4())
outputs_to_execute = valid[2] outputs_to_execute = valid[2]
routing_key = method.routing_key
task_info = { task_info = {
"workflow": workflow, "workflow": workflow,
"prompt_id": prompt_id, "prompt_id": prompt_id,
@ -431,8 +431,8 @@ class MemedeckWorker:
return return
api_endpoint = '/generation/update' if task['workflow'] != 'training' else '/training/update' 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]: sending to api: {api_endpoint}")
self.logger.info(f"[memedeck]: data: {data}") # self.logger.info(f"[memedeck]: data: {data}")
try: try:
# this request is not sending properly for faceswap # this request is not sending properly for faceswap
post_func = partial(requests.post, f"{self.api_url}{api_endpoint}", json=data) post_func = partial(requests.post, f"{self.api_url}{api_endpoint}", json=data)