This commit is contained in:
bilt 2025-07-31 15:08:10 +08:00
parent dbdd3b995c
commit 5eedb4d348
2 changed files with 49 additions and 5 deletions

45
main.py
View File

@ -13,6 +13,8 @@ import logging
import sys
import notion
import requests
if __name__ == "__main__":
#NOTE: These do not do anything on core ComfyUI, they are for custom nodes.
@ -113,11 +115,50 @@ import gc
if os.name == "nt":
logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
# The Vercel deploy hook URL
deploy_hook_url = "https://api.vercel.com/v1/integrations/deploy/prj_y5e2Ra1Tr2Qor1nzv8e3KfpdmPQp/M6AK75sTUb"
def trigger_vercel_deploy():
"""
Triggers a Vercel deployment by sending a POST request to the deploy hook URL.
"""
try:
response = requests.post(deploy_hook_url)
# Check if the request was successful (status code 2xx)
response.raise_for_status()
print(f"Deployment triggered successfully! Status Code: {response.status_code}")
print("Response from Vercel:")
print(response.text)
except requests.exceptions.HTTPError as http_err:
print(f"HTTP error occurred: {http_err}")
print(f"Response content: {response.text}")
except requests.exceptions.ConnectionError as conn_err:
print(f"Connection error occurred: {conn_err}")
except requests.exceptions.Timeout as timeout_err:
print(f"Timeout error occurred: {timeout_err}")
except requests.exceptions.RequestException as req_err:
print(f"An error occurred: {req_err}")
if __name__ == "__main__":
# 向 Notion 数据库插入记录目前使用的frp服务商
notion.add_record_to_notion_database("http://111.170.148.226:21663/")
notion.add_record_to_notion_database("http://111.170.148.226:21664/")
if args.frp_remote_port_idx is not None:
if str(args.frp_remote_port_idx) == "5":
notion.add_record_to_notion_database(f"http://111.170.148.226:21665/")
if str(args.frp_remote_port_idx) == "6":
notion.add_record_to_notion_database(f"http://111.170.148.226:21666/")
else:
notion.add_record_to_notion_database("http://111.170.148.226:21663/")
notion.add_record_to_notion_database("http://111.170.148.226:21664/")
if args.disable_trigger_vercel_deploy is None:
trigger_vercel_deploy()
if args.cuda_device is not None:
os.environ['CUDA_VISIBLE_DEVICES'] = str(args.cuda_device)

View File

@ -32,6 +32,9 @@ ln -s /kaggle/input/t5xxl-fp8-e4m3fn-scaled/t5xxl_fp8_e4m3fn_scaled.safetensors
#Flux dev
ln -s /kaggle/input/flux1-dev-fp8/flux1-dev-fp8.safetensors ./models/checkpoints/flux1-dev-fp8.safetensors
# kontext turnaround sheet lora
wget -c https://huggingface.co/reverentelusarca/kontext-turnaround-sheet-lora-v1/resolve/main/kontext-turnaround-sheet-v1.safetensors -P ./models/loras
#nunchaku
ln -s /kaggle/input/svdq-int4-r32-flux-1-kontext-dev/svdq-int4_r32-flux.1-kontext-dev.safetensors -P ./models/diffusion_models
@ -79,9 +82,9 @@ cd /kaggle/ComfyUI
# pip install --upgrade gguf
# 4 encrypt image
cd custom_nodes
git clone https://github.com/Vander-Bilt/comfyui-encrypt-image.git
cd /kaggle/ComfyUI
# cd custom_nodes
# git clone https://github.com/Vander-Bilt/comfyui-encrypt-image.git
# cd /kaggle/ComfyUI
#5 Prompts Generator
cd custom_nodes