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."