#!/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"