update

download_checkpoints.sh

@@ -47,7 +47,9 @@ if [ -d "checkpoints/videomama_unet" ]; then
 else
     git clone https://huggingface.co/SammyLim/VideoMama checkpoints/videomama
 fi
+echo "videomama checkpoint setup complete."
 
+echo "Checking Stable Video Diffusion Img2Vid XT checkpoint..."
 git clone https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt checkpoints/stable-video-diffusion-img2vid-xt
 
 echo ""

videomama_wrapper.py

@@ -3,6 +3,15 @@ VideoMaMa Inference Wrapper
 Handles video matting with mask conditioning
 """
 
+import os
+import torch
+
+# CRITICAL: Redirect cache to temporary storage
+os.environ['TORCH_HOME'] = '/tmp/torch_cache'
+os.environ['HUB_DIR'] = '/tmp/torch_hub'
+os.environ['TMPDIR'] = '/tmp'
+torch.hub.set_dir('/tmp/torch_hub')
+
 import os
 import torch
 import numpy as np