From 8732d5d6fe9950b5ffca77b7b808c4b93f9894dc Mon Sep 17 00:00:00 2001
From: bilt <vanderbilt@gmail.com>
Date: Tue, 19 Aug 2025 08:46:17 +0800
Subject: [PATCH] sigterm,  mylora

---
 main.py   | 8 ++++++++
 script.sh | 4 ++++
 2 files changed, 12 insertions(+)

diff --git a/main.py b/main.py
index 59249913c..cf9c487ed 100644
--- a/main.py
+++ b/main.py
@@ -17,6 +17,14 @@ import requests
 
 
 if __name__ == "__main__":
+    import signal
+    import sys
+
+    def sigterm_handler(_signo, _stack_frame):
+        logging.info("Kaggle session terminated. Shutting down gracefully.")
+        sys.exit(0)
+
+    signal.signal(signal.SIGTERM, sigterm_handler)
     #NOTE: These do not do anything on core ComfyUI, they are for custom nodes.
     os.environ['HF_HUB_DISABLE_TELEMETRY'] = '1'
     os.environ['DO_NOT_TRACK'] = '1'
diff --git a/script.sh b/script.sh
index 80ed059bb..78b3bacc9 100644
--- a/script.sh
+++ b/script.sh
@@ -3,6 +3,10 @@
 # cosmos ？
 ln -s /kaggle/input/cosmos-predict2-2b-video2world-480p-16fps/cosmos_predict2_2B_video2world_480p_16fps.safetensors ./models/diffusion_models/cosmos_predict2_2B_video2world_480p_16fps.safetensors
 
+# my lora
+git clone https://huggingface.co/datasets/Heng365/outputs /kaggle/working/fromhf
+mv /kaggle/working/fromhf/* ./models/loras
+
 # sd lora ？
 ln -s /kaggle/input/moxinv1/MoXinV1.safetensors ./models/loras/MoXinV1.safetensors
 ln -s /kaggle/input/blindbox-v1-mix/blindbox_v1_mix.safetensors ./models/loras/blindbox_v1_mix.safetensors