sigterm, mylora

This commit is contained in:
bilt 2025-08-19 08:46:17 +08:00
parent 1da8bdbcd4
commit 8732d5d6fe
2 changed files with 12 additions and 0 deletions

View File

@ -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'

View File

@ -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