runtime_env.yaml 224 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
working_dir: ./
excludes: ["/.git/"]
env_vars:
chenych's avatar
update  
chenych committed
4
5
6
  TOKENIZERS_PARALLELISM: "true"
  NCCL_DEBUG: "WARN"
  VLLM_LOGGING_LEVEL: "INFO"
chenych's avatar
chenych committed
7
  TORCH_NCCL_AVOID_RECORD_STREAMS: "1"
chenych's avatar
update  
chenych committed
8
  PYTORCH_CUDA_ALLOC_CONF: "expandable_segments:False"