chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: CD_BYPASS_UPLOAD

Browse Source
This commit is contained in:
bennykok 2024-05-10 11:34:49 +09:00
parent 697fd52349
commit 00d827e232
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
custom_routes.py
View File

@ -30,6 +30,10 @@ api_task = None
cd_enable_log = os.environ.get('CD_ENABLE_LOG', 'false').lower() == 'true' cd_enable_log = os.environ.get('CD_ENABLE_LOG', 'false').lower() == 'true'
cd_enable_run_log = os.environ.get('CD_ENABLE_RUN_LOG', 'false').lower() == 'true' cd_enable_run_log = os.environ.get('CD_ENABLE_RUN_LOG', 'false').lower() == 'true'
bypass_upload = os.environ.get('CD_BYPASS_UPLOAD', 'false').lower() == 'true'
print("CD_BYPASS_UPLOAD", bypass_upload)
def clear_current_prompt(sid): def clear_current_prompt(sid):
prompt_server = server.PromptServer.instance prompt_server = server.PromptServer.instance
@ -974,19 +978,19 @@ async def update_run_with_output(prompt_id, data, node_id=None):
"output_data": data "output_data": data
} }
try: if not bypass_upload:
have_upload = 'images' in data or 'files' in data or 'gifs' in data or 'mesh' in data try:
print("\nhave_upload", have_upload, node_id) have_upload = 'images' in data or 'files' in data or 'gifs' in data or 'mesh' in data
print("\nhave_upload", have_upload, node_id)
if have_upload: if have_upload:
await update_file_status(prompt_id, data, True, node_id=node_id) await update_file_status(prompt_id, data, True, node_id=node_id)
# asyncio.create_task(upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload)) # asyncio.create_task(upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload))
await upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload) await upload_in_background(prompt_id, data, node_id=node_id, have_upload=have_upload)
except Exception as e: except Exception as e:
await handle_error(prompt_id, data, e) await handle_error(prompt_id, data, e)
requests.post(status_endpoint, json=body) requests.post(status_endpoint, json=body)