mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-15 21:36:42 +08:00
sigterm, mylora
This commit is contained in:
parent
1da8bdbcd4
commit
8732d5d6fe
8
main.py
8
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'
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user