File size: 499 Bytes
b39ee59 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | #!/usr/bin/env bash
set -e
mkdir -p /workspace/cache/huggingface
mkdir -p /workspace/cache/rembg
mkdir -p /workspace/cache/torch
export HF_HOME=/workspace/cache/huggingface
export TRANSFORMERS_CACHE=/workspace/cache/huggingface
export DIFFUSERS_CACHE=/workspace/cache/huggingface
export U2NET_HOME=/workspace/cache/rembg
export TORCH_HOME=/workspace/cache/torch
echo "RunPod cache environment configured:"
echo " HF_HOME=$HF_HOME"
echo " U2NET_HOME=$U2NET_HOME"
echo " TORCH_HOME=$TORCH_HOME"
|