File size: 1,033 Bytes
2ea2835 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | #!/usr/bin/env bash
set -euo pipefail
REPRO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
source "$REPRO_ROOT/configs/local.env"
PYTHON_BIN="$(command -v "${PYTHON_BIN:-python}")"
CONFIG="$REPRO_ROOT/configs/local.yaml"
export SELF_FORCING_ASSET_ROOT="$VIDEO_GENERATION_ASSET_ROOT"
export SELF_FORCING_MODELS_DIR="$VIDEO_GENERATION_ASSET_ROOT/models"
export SELF_FORCING_CKPT_DIR="$VIDEO_GENERATION_ASSET_ROOT/checkpoints"
export PYTHONPATH="$REPRO_ROOT/code${PYTHONPATH:+:$PYTHONPATH}"
export PYTHONUNBUFFERED=1 PYTHONDONTWRITEBYTECODE=1
export OMP_NUM_THREADS="${OMP_NUM_THREADS:-12}" TOKENIZERS_PARALLELISM=false
export HF_HOME="${HF_HOME:-$REPRO_ROOT/cache/huggingface}"
export TORCH_HOME="${TORCH_HOME:-$REPRO_ROOT/cache/torch}"
export TRITON_CACHE_DIR="${TRITON_CACHE_DIR:-$REPRO_ROOT/cache/triton}"
export HF_HUB_OFFLINE=1 TRANSFORMERS_OFFLINE=1
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export TORCH_NCCL_ASYNC_ERROR_HANDLING=1 NCCL_DEBUG=WARN
unset WORLD_SIZE RANK LOCAL_RANK MASTER_ADDR MASTER_PORT
|