From 214c43d62938c72032bd00324654f9c8fab56b6c Mon Sep 17 00:00:00 2001
From: bilt <vanderbilt@gmail.com>
Date: Thu, 31 Jul 2025 15:10:29 +0800
Subject: [PATCH] tests

---
 script.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/script.sh b/script.sh
index 39304311f..a45b7c273 100644
--- a/script.sh
+++ b/script.sh
@@ -120,12 +120,18 @@ tar -xzvf /kaggle/working/frp_0.54.0_linux_amd64.tar.gz -C /kaggle/working
 cp -p /kaggle/working/frp_0.54.0_linux_amd64/frpc /kaggle/working/frpc
 cp -p /kaggle/ComfyUI/template_frpc /kaggle/working/frpc.toml
 
+# 1, 2 主要是为了兼容之前的comfyUI notebook（不想一个一个的去修改了）
 FRP_CONFIG_FILE="/kaggle/working/frpc.toml"
 CHOICE="$1"
 if [ "$CHOICE" -eq 1 ]; then
   TARGET_REMOTE_PORT="21663"
 elif [ "$CHOICE" -eq 2 ]; then
   TARGET_REMOTE_PORT="21664"
+
+elif [ "$CHOICE" -eq 5 ]; then
+  TARGET_REMOTE_PORT="21665"
+elif [ "$CHOICE" -eq 6 ]; then
+  TARGET_REMOTE_PORT="21666"
 else
   echo "Invalid CHOICE: $CHOICE"
   echo "Only 1 or 2 are supported."